Unverified Commit cb757348 authored by yfzhou's avatar yfzhou Committed by GitHub
Browse files

【bug】reload altclip model error 

When using BertSeriesModelWithTransformation as the cond_stage_model, the undo_hijack should be performed using the FrozenXLMREmbedderWithCustomWords type; otherwise, it will result in a failed model reload.
parent f865d3e1

modules/sd_hijack.py

+1 −1
Original line number Diff line number Diff line
@@ -203,7 +203,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:

CRA Git | Maintained and supported by SUSTech CRA and CCSE