Support loading old CLIP models saved with CLIPSave.

MoonRide303 · Feb 25, 2024 · ca7c310 · ca7c310
1 parent 8d7910c
commit ca7c310
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/comfy/sd.py b/comfy/sd.py
@@ -365,6 +365,9 @@ class EmptyClass:
     for i in range(len(clip_data)):
         if "transformer.resblocks.0.ln_1.weight" in clip_data[i]:
             clip_data[i] = comfy.utils.clip_text_transformers_convert(clip_data[i], "", "")
+        else:
+            if "text_projection" in clip_data[i]:
+                clip_data[i]["text_projection.weight"] = clip_data[i]["text_projection"].transpose(0, 1) #old models saved with the CLIPSave node
 
     clip_target = EmptyClass()
     clip_target.params = {}