mirror/stable-diffusion-webui
2
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-11-27 06:40:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix broken img2img

Browse Source
This commit is contained in:
AUTOMATIC1111 2023-07-13 16:18:39 +03:00
parent 76ebb175ca
commit 6f23da603d
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/sd_models_xl.py
View File

@ -32,6 +32,9 @@ def apply_model(self: sgm.models.diffusion.DiffusionEngine, x, t, cond):
return self.model(x, t, cond) return self.model(x, t, cond)
def get_first_stage_encoding(self, x): # SDXL's encode_first_stage does everything so get_first_stage_encoding is just there for compatibility
return x
def extend_sdxl(model): def extend_sdxl(model):
dtype = next(model.model.diffusion_model.parameters()).dtype dtype = next(model.model.diffusion_model.parameters()).dtype
model.model.diffusion_model.dtype = dtype model.model.diffusion_model.dtype = dtype
@ -50,6 +53,7 @@ def extend_sdxl(model):
sgm.models.diffusion.DiffusionEngine.get_learned_conditioning = get_learned_conditioning sgm.models.diffusion.DiffusionEngine.get_learned_conditioning = get_learned_conditioning
sgm.models.diffusion.DiffusionEngine.apply_model = apply_model sgm.models.diffusion.DiffusionEngine.apply_model = apply_model
sgm.models.diffusion.DiffusionEngine.get_first_stage_encoding = get_first_stage_encoding
sgm.modules.attention.print = lambda *args: None sgm.modules.attention.print = lambda *args: None
sgm.modules.diffusionmodules.model.print = lambda *args: None sgm.modules.diffusionmodules.model.print = lambda *args: None