AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

debug

Browse Source
This commit is contained in:
sayakpaul
2023-11-14 11:12:15 +05:30
parent 3b066d2657
commit fc5546fecc
2 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/consistency_distillation/train_lcm_distill_lora_sdxl.py
View File

@@ -136,7 +136,7 @@ def log_validation(vae, unet, args, accelerator, weight_dtype, step):
diffusers_state_dict = {
f"{pipeline.unet_name}.{module_name}": param for module_name, param in diffusers_state_dict.items()
}
print(list(diffusers_state_dict.keys()))
pipeline.load_lora_weights(diffusers_state_dict)
pipeline.fuse_lora()

1
src/diffusers/utils/state_dict_utils.py
View File

@@ -210,7 +210,6 @@ def convert_state_dict_to_diffusers(state_dict, original_type=None, **kwargs):
if original_type not in DIFFUSERS_STATE_DICT_MAPPINGS.keys():
raise ValueError(f"Original type {original_type} is not supported")
print(f"*******From state_dict_utils.py: {original_type}*******")
mapping = DIFFUSERS_STATE_DICT_MAPPINGS[original_type]
return convert_state_dict(state_dict, mapping)