fix

src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py

@@ -446,8 +446,9 @@ def convert_ldm_unet_checkpoint(
         new_checkpoint["add_embedding.linear_2.bias"] = unet_state_dict["label_emb.0.2.bias"]
 
     # Relevant to StableDiffusionUpscalePipeline
-    if (config["num_class_embeds"] is not None) and ("label_emb.weight" in unet_state_dict):
-        new_checkpoint["class_embedding.weight"] = unet_state_dict["label_emb.weight"]
+    if "num_class_embeds" in config:
+        if (config["num_class_embeds"] is not None) and ("label_emb.weight" in unet_state_dict):
+            new_checkpoint["class_embedding.weight"] = unet_state_dict["label_emb.weight"]
 
     new_checkpoint["conv_in.weight"] = unet_state_dict["input_blocks.0.0.weight"]
     new_checkpoint["conv_in.bias"] = unet_state_dict["input_blocks.0.0.bias"]