Move silu to sd_hijack

This commit is contained in:
Jairo Correa
2022-09-29 01:14:54 -03:00
parent c938679de7
commit c2d5b29040
2 changed files with 3 additions and 12 deletions

View File

@@ -22,10 +22,7 @@ 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()