Remove duplicate text_projection key from some saved models.

37a86e46 · comfyanonymous · 8daedc5b · 37a86e46
Commit 37a86e46 authored Feb 28, 2024 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

comfy/diffusers_convert.py comfy/diffusers_convert.py +3 -3

No files found.
--- a/comfy/diffusers_convert.py
+++ b/comfy/diffusers_convert.py
@@ -240,7 +240,7 @@ def convert_text_enc_state_dict_v20(text_enc_dict, prefix=""):
        text_proj = "transformer.text_projection.weight"
        if k.endswith(text_proj):
            new_state_dict[k.replace(text_proj, "text_projection")] = v.transpose(0, 1).contiguous()
-
+        else:
            relabelled_key = textenc_pattern.sub(lambda m: protected[re.escape(m.group(0))], k)
            new_state_dict[relabelled_key] = v