Support loading clip_g from diffusers in CLIP Loader nodes.

2 years ago · d3f5998218
parent a9a4ba7574
commit d3f5998218
1 changed files with 2 additions and 0 deletions
--- a/comfy/sdxl_clip.py
+++ b/comfy/sdxl_clip.py
@ -35,6 +35,8 @@ class SDXLClipG(sd1_clip.SD1ClipModel):
    def load_sd(self, sd):
        if "text_projection" in sd:
            self.text_projection[:] = sd.pop("text_projection")
+        if "text_projection.weight" in sd:
+            self.text_projection[:] = sd.pop("text_projection.weight").transpose(0, 1)
        return super().load_sd(sd)

 class SDXLClipGTokenizer(sd1_clip.SD1Tokenizer):