Fix issue with some models.

0108616b · comfyanonymous · 2973ff24 · 0108616b
Commit 0108616b authored Jan 28, 2023 by comfyanonymous
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 2 deletions

comfy/sd.py comfy/sd.py +5 -2

No files found.
--- a/comfy/sd.py
+++ b/comfy/sd.py
@@ -32,8 +32,11 @@ def load_model_from_config(config, ckpt, verbose=False, load_state_dict_to=[]):
            y = x.replace("cond_stage_model.transformer.", "cond_stage_model.transformer.text_model.")
            sd[y] = sd.pop(x)

-    if 'cond_stage_model.transformer.text_model.embeddings.position_ids' in sd:
-        sd['cond_stage_model.transformer.text_model.embeddings.position_ids'] = sd['cond_stage_model.transformer.text_model.embeddings.position_ids'].round()
+    try:
+        if 'cond_stage_model.transformer.text_model.embeddings.position_ids' in sd:
+            sd['cond_stage_model.transformer.text_model.embeddings.position_ids'] = sd['cond_stage_model.transformer.text_model.embeddings.position_ids'].round()
+    except:
+        pass

    for x in load_state_dict_to:
        x.load_state_dict(sd, strict=False)