fffiloni commited on
Commit
e2c1f58
·
verified ·
1 Parent(s): 401ccdf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -56,8 +56,8 @@ def main(image_path, b_lora_trained_folder, instance_prompt):
56
  shutil.copy(image_path, local_dir)
57
  print(f"source image has been copied in {local_dir} directory")
58
 
59
- max_train_steps = 10
60
- checkpoint_steps = 10
61
 
62
  train_dreambooth_blora_sdxl(local_dir, b_lora_trained_folder, instance_prompt, max_train_steps, checkpoint_steps)
63
 
 
56
  shutil.copy(image_path, local_dir)
57
  print(f"source image has been copied in {local_dir} directory")
58
 
59
+ max_train_steps = 1000
60
+ checkpoint_steps = 500
61
 
62
  train_dreambooth_blora_sdxl(local_dir, b_lora_trained_folder, instance_prompt, max_train_steps, checkpoint_steps)
63