Merge pull request #11878 from Bourne-M/patch-1
【bug】reload altclip model error
This commit is contained in:
parent
2c11e9009e
commit
c5934fb6e3
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue