Skip to content

GitLab

Commit f90a5139 authored by Dhruv Nair's avatar Dhruv Nair
Browse files

fix

parent a2bc2e14
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
View file @ f90a5139
...@@ -446,6 +446,7 @@ def convert_ldm_unet_checkpoint( ...@@ -446,6 +446,7 @@ def convert_ldm_unet_checkpoint(
new_checkpoint["add_embedding.linear_2.bias"] = unet_state_dict["label_emb.0.2.bias"] new_checkpoint["add_embedding.linear_2.bias"] = unet_state_dict["label_emb.0.2.bias"]
# Relevant to StableDiffusionUpscalePipeline # Relevant to StableDiffusionUpscalePipeline
if "num_class_embeds" in config:
if (config["num_class_embeds"] is not None) and ("label_emb.weight" in unet_state_dict): 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["class_embedding.weight"] = unet_state_dict["label_emb.weight"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment