Hugging Face's logo

Spaces:
fffiloni
/
B-LoRa-trainer
Running

App Files Community
1
fffiloni commited on
Commit
95d2be4
·
verified ·
1 Parent(s): c32685e

Update app.py

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