diff --git a/examples/consistency_distillation/train_lcm_distill_lora_sdxl.py b/examples/consistency_distillation/train_lcm_distill_lora_sdxl.py index 3885001915..7c59a5c94d 100644 --- a/examples/consistency_distillation/train_lcm_distill_lora_sdxl.py +++ b/examples/consistency_distillation/train_lcm_distill_lora_sdxl.py @@ -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() diff --git a/src/diffusers/utils/state_dict_utils.py b/src/diffusers/utils/state_dict_utils.py index 5f6d1df5dd..777c611f71 100644 --- a/src/diffusers/utils/state_dict_utils.py +++ b/src/diffusers/utils/state_dict_utils.py @@ -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