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

Fix keys for lora format on advanced training scripts (#6361)

Browse Source 
fix keys for lora format on advanced training scripts
This commit is contained in:
apolinário
2023-12-27 11:38:03 -06:00
committed by GitHub
parent 645a62bf3b
commit 1fff527702
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py
View File

@@ -1955,7 +1955,7 @@ def main(args):
if accelerator.is_main_process:
unet = accelerator.unwrap_model(unet)
unet = unet.to(torch.float32)
unet_lora_layers = get_peft_model_state_dict(unet)
unet_lora_layers = convert_state_dict_to_diffusers(get_peft_model_state_dict(unet))
if args.train_text_encoder:
text_encoder_one = accelerator.unwrap_model(text_encoder_one)