ysharma HF staff commited on
Commit
b12d0df
·
1 Parent(s): 7722e4b
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -41,9 +41,9 @@ def accelerate_train_lora(steps):
41
  #subprocess.run(accelerate launch {"./train_lora_dreambooth.py"} \
42
  #subprocess.Popen(f'accelerate launch {"./train_lora_dreambooth.py"} \
43
  os.system( f'accelerate launch {"./train_lora_dreambooth.py"} \
44
- --pretrained_model_name_or_path=MODEL_NAME \
45
- --instance_data_dir=INSTANCE_DIR \
46
- --output_dir=OUTPUT_DIR \
47
  --instance_prompt="style of sks" \
48
  --resolution=512 \
49
  --train_batch_size=1 \
 
41
  #subprocess.run(accelerate launch {"./train_lora_dreambooth.py"} \
42
  #subprocess.Popen(f'accelerate launch {"./train_lora_dreambooth.py"} \
43
  os.system( f'accelerate launch {"./train_lora_dreambooth.py"} \
44
+ --pretrained_model_name_or_path={MODEL_NAME} \
45
+ --instance_data_dir={INSTANCE_DIR} \
46
+ --output_dir={OUTPUT_DIR} \
47
  --instance_prompt="style of sks" \
48
  --resolution=512 \
49
  --train_batch_size=1 \