Support loading clip_g from diffusers in CLIP Loader nodes.

im-stuff · Jul 9, 2023 · d3f5998 · d3f5998
1 parent a9a4ba7
commit d3f5998
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/comfy/sdxl_clip.py b/comfy/sdxl_clip.py
@@ -35,6 +35,8 @@ def clip_layer(self, layer_idx):
     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):