Skip to content

GitLab

Unverified Commit 980736b7 authored by hlky's avatar hlky Committed by GitHub
Browse files

Fix train_dreambooth_lora_sd3_miniature (#10554)

parent 50c81df4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/research_projects/sd3_lora_colab/train_dreambooth_lora_sd3_miniature.py
View file @ 980736b7
...@@ -765,7 +765,7 @@ def main(args): ...@@ -765,7 +765,7 @@ def main(args):
lora_state_dict = StableDiffusion3Pipeline.lora_state_dict(input_dir) lora_state_dict = StableDiffusion3Pipeline.lora_state_dict(input_dir)
transformer_state_dict = { transformer_state_dict = {
f'{k.replace("transformer.", "")}': v for k, v in lora_state_dict.items() if k.startswith("unet.") f'{k.replace("transformer.", "")}': v for k, v in lora_state_dict.items() if k.startswith("transformer.")
} }
transformer_state_dict = convert_unet_state_dict_to_peft(transformer_state_dict) transformer_state_dict = convert_unet_state_dict_to_peft(transformer_state_dict)
incompatible_keys = set_peft_model_state_dict(transformer_, transformer_state_dict, adapter_name="default") incompatible_keys = set_peft_model_state_dict(transformer_, transformer_state_dict, adapter_name="default")
......
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