Skip to content

Commit 8f21a9f

Browse files
Fix convert SD to diffusers error (#1979) (#2529)
Fix convert sd 768 error (#1979) Co-authored-by: tweeter0830 <tweeter0830@users.noreply.github.com>
1 parent d9b9533 commit 8f21a9f

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -841,7 +841,10 @@ def convert_open_clip_checkpoint(checkpoint):
841841

842842
text_model_dict = {}
843843

844-
d_model = int(checkpoint["cond_stage_model.model.text_projection"].shape[0])
844+
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
845848

846849
text_model_dict["text_model.embeddings.position_ids"] = text_model.text_model.embeddings.get_buffer("position_ids")
847850

0 commit comments

Comments
 (0)