fix: checkpoint save issue in advanced dreambooth lora sdxl script (#8926)

Co-authored-by: Linoy Tsaban <57615435+linoytsaban@users.noreply.github.com>

examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py

@@ -1605,10 +1605,12 @@ def main(args):
                 if isinstance(model, type(unwrap_model(unet))):
                     unet_lora_layers_to_save = convert_state_dict_to_diffusers(get_peft_model_state_dict(model))
                 elif isinstance(model, type(unwrap_model(text_encoder_one))):
+                    if args.train_text_encoder:
                         text_encoder_one_lora_layers_to_save = convert_state_dict_to_diffusers(
                             get_peft_model_state_dict(model)
                         )
                 elif isinstance(model, type(unwrap_model(text_encoder_two))):
+                    if args.train_text_encoder:
                         text_encoder_two_lora_layers_to_save = convert_state_dict_to_diffusers(
                             get_peft_model_state_dict(model)
                         )