Update app.py

app.py

@@ -119,13 +119,13 @@ def train_dreambooth_lora_sdxl(dataset_id, instance_data_dir, lora_trained_xl_fo
         f"--output_dir={lora_trained_xl_folder}",
         "--mixed_precision=fp16",
         f"--instance_prompt={instance_prompt}",
-        "--resolution=512",
-        "--train_batch_size=4",
+        "--resolution=1024",
+        "--train_batch_size=2",
         "--gradient_accumulation_steps=2",
         "--gradient_checkpointing",
-        "--learning_rate=1e-6",
+        "--learning_rate=1e-4",
         "--lr_scheduler=constant",
-        "--lr_warmup_steps=100",
+        "--lr_warmup_steps=0",
         "--enable_xformers_memory_efficient_attention",
         "--mixed_precision=fp16",
         "--use_8bit_adam",