mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-01-21 12:02:32 -08:00
Code Issues Projects Releases Packages Wiki Activity

add hash and fix undo hijack bug

Browse source 
Signed-off-by: zhaohu xing <920232796@qq.com>
This commit is contained in:
zhaohu xing 2022-12-06 16:04:50 +08:00
parent a25dfebeed
commit 5dcc22606d
4 changed files with 11 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

6
modules/sd_hijack.py
View file

@ -112,7 +112,11 @@ class StableDiffusionModelHijack:
self.layers = flatten(m)
def undo_hijack(self, m):
if type(m.cond_stage_model) == sd_hijack_clip.FrozenCLIPEmbedderWithCustomWords:
if shared.text_model_name == "XLMR-Large":
m.cond_stage_model = m.cond_stage_model.wrapped
elif type(m.cond_stage_model) == sd_hijack_clip.FrozenCLIPEmbedderWithCustomWords:
m.cond_stage_model = m.cond_stage_model.wrapped
model_embeddings = m.cond_stage_model.transformer.text_model.embeddings