fix "File does not contain tensor conditioner.embedders.0.transformer…

….text_model.embeddings.position_ids" error with SDXL models issue #136 reported by @CCpt5
wkpark · Apr 13, 2024 · 1701dd3 · 1701dd3
1 parent efc73c0
commit 1701dd3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/scripts/model_mixer.py b/scripts/model_mixer.py
@@ -3804,7 +3804,11 @@ def dare_merge(theta0, theta1, alpha, p):
         sel_keys = changed_keys if partial_update else keys
 
         # save some dicts
-        checkpoint_dict_skip_on_merge = ["cond_stage_model.transformer.text_model.embeddings.position_ids", "conditioner.embedders.1.model.transformer.text_model.embeddings.position_ids" ]
+        checkpoint_dict_skip_on_merge = [
+            "cond_stage_model.transformer.text_model.embeddings.position_ids",
+            "conditioner.embedders.0.transformer.text_model.embeddings.position_ids",
+            "conditioner.embedders.1.model.transformer.text_model.embeddings.position_ids",
+        ]
         for k in checkpoint_dict_skip_on_merge:
             if k in sel_keys:
                 sel_keys.remove(k)