We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d9b9533 commit 8f21a9fCopy full SHA for 8f21a9f
src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
@@ -841,7 +841,10 @@ def convert_open_clip_checkpoint(checkpoint):
841
842
text_model_dict = {}
843
844
- d_model = int(checkpoint["cond_stage_model.model.text_projection"].shape[0])
+ if "cond_stage_model.model.text_projection" in checkpoint:
845
+ d_model = int(checkpoint["cond_stage_model.model.text_projection"].shape[0])
846
+ else:
847
+ d_model = 1024
848
849
text_model_dict["text_model.embeddings.position_ids"] = text_model.text_model.embeddings.get_buffer("position_ids")
850
0 commit comments