Merge pull request #11878 from Bourne-M/patch-1

【bug】reload altclip model error

Merge pull request #11878 from Bourne-M/patch-1
【bug】reload altclip model error
c5934fb6 · AUTOMATIC1111 · 2c11e900 · c5934fb6
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

modules/sd_hijack.py modules/sd_hijack.py +1 -1

未找到文件。
--- a/modules/sd_hijack.py
+++ b/modules/sd_hijack.py
@@ -243,7 +243,7 @@ class StableDiffusionModelHijack:
        ldm.modules.diffusionmodules.openaimodel.UNetModel.forward = sd_unet.UNetModel_forward

    def undo_hijack(self, m):
-        if type(m.cond_stage_model) == xlmr.BertSeriesModelWithTransformation:
+        if type(m.cond_stage_model) == sd_hijack_xlmr.FrozenXLMREmbedderWithCustomWords:
            m.cond_stage_model = m.cond_stage_model.wrapped

        elif type(m.cond_stage_model) == sd_hijack_clip.FrozenCLIPEmbedderWithCustomWords: