ysharma HF Staff commited on
Commit
132f75e
·
1 Parent(s): d357dc9

update training code

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ def accelerate_train_lora(steps, images, in_prompt):
65
  --pretrained_model_name_or_path={MODEL_NAME} \
66
  --instance_data_dir={INSTANCE_DIR} \
67
  --output_dir={OUTPUT_DIR} \
68
- --instance_prompt={in_prompt} \
69
  --train_text_encoder \
70
  --resolution=512 \
71
  --train_batch_size=1 \
 
65
  --pretrained_model_name_or_path={MODEL_NAME} \
66
  --instance_data_dir={INSTANCE_DIR} \
67
  --output_dir={OUTPUT_DIR} \
68
+ --instance_prompt="""{in_prompt}""" \
69
  --train_text_encoder \
70
  --resolution=512 \
71
  --train_batch_size=1 \