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

remove prints.

Browse Source
This commit is contained in:
sayakpaul
2023-11-14 11:23:33 +05:30
parent 35e30fbb57
commit 53c13f7f8d
2 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/consistency_distillation/train_lcm_distill_lora_sdxl.py
View File

@@ -136,7 +136,6 @@ def log_validation(vae, unet, args, accelerator, weight_dtype, step):
diffusers_state_dict = {
f"{module_name.replace('base_model.model', 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

@@ -137,7 +137,6 @@ def convert_state_dict(state_dict, mapping):
k = k.replace(pattern, new_pattern)
break
converted_state_dict[k] = v
print("From state_dict utils", any("lora_A" in k for k in converted_state_dict))
return converted_state_dict