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

fix keys for lora format on advanced training scripts

Author: apolinario

examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py

@@ -1955,7 +1955,7 @@ def compute_text_embeddings(prompt, text_encoders, tokenizers):
     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)