Browse Source

Fix issue with some models.

pull/3/head
comfyanonymous 2 years ago
parent
commit
0108616b77
  1. 7
      comfy/sd.py

7
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)

Loading…
Cancel
Save