Update app.py

app.py

@@ -112,8 +112,8 @@ def train_dreambooth_lora_sdxl(dataset_id, instance_data_dir, lora_trained_xl_fo
         "accelerate",
         "launch",
         script_filename,  # Use the local script
-        "--pretrained_model_name_or_path=runwayml/stable-diffusion-v1-5",
-        "--pretrained_vae_model_name_or_path=stabilityai/sd-vae-ft-mse",
+        "--pretrained_model_name_or_path=stabilityai/stable-diffusion-xl-base-1.0",
+        "--pretrained_vae_model_name_or_path=madebyollin/sdxl-vae-fp16-fix",
         f"--dataset_id={dataset_id}",
         f"--instance_data_dir={instance_data_dir}",
         f"--output_dir={lora_trained_xl_folder}",