fix to work with .ckpt from comfyui

kohya-ss · Jul 5, 2023 · 4a34e58 · 4a34e58
1 parent 3d0375d
commit 4a34e58
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/library/sdxl_model_util.py b/library/sdxl_model_util.py
@@ -44,6 +44,9 @@ def convert_key(key):
             key = key.replace(".token_embedding.weight", ".embeddings.token_embedding.weight")
         elif ".ln_final" in key:
             key = key.replace(".ln_final", ".final_layer_norm")
+        # ckpt from comfy has this key: text_model.encoder.text_model.embeddings.position_ids
+        elif ".embeddings.position_ids" in key:
+            key = None  # remove this key: make position_ids by ourselves
         return key
 
     keys = list(checkpoint.keys())