Fix memory leak and reduce memory usage

This commit is contained in:
Jairo Correa
2022-09-28 22:14:13 -03:00
parent 041d2aefc0
commit c938679de7
6 changed files with 42 additions and 16 deletions

View File

@@ -22,7 +22,10 @@ import modules.txt2img
import modules.img2img
import modules.swinir as swinir
import modules.sd_models
from torch.nn.functional import silu
import ldm
ldm.modules.diffusionmodules.model.nonlinearity = silu
modules.codeformer_model.setup_codeformer()
modules.gfpgan_model.setup_gfpgan()