dkebudi commited on
Commit
07a1fda
·
verified ·
1 Parent(s): 8352565

fix for crop_center variable (#4)

Browse files

- fix for crop_center variable (9a6b469099bb500a69e49a82803a887068f6d38e)

Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -209,6 +209,7 @@ def start_training(
209
  text_encoder_learning_rate,
210
  seed,
211
  resolution,
 
212
  num_train_epochs,
213
  checkpointing_steps,
214
  prior_loss_weight,
@@ -252,7 +253,7 @@ def start_training(
252
  f"output_dir={slugged_lora_name}",
253
  f"mixed_precision={mixed_precision}",
254
  f"resolution={int(resolution)}",
255
- f"center_crop={bool(True)}",
256
  f"train_batch_size={int(train_batch_size)}",
257
  f"repeats={int(repeats)}",
258
  f"gradient_accumulation_steps={int(gradient_accumulation_steps)}",
@@ -435,7 +436,7 @@ def start_training_og(
435
  f"--output_dir={slugged_lora_name}",
436
  f"--mixed_precision={mixed_precision}",
437
  f"--resolution={int(resolution)}",
438
- f"--center_crop={bool(True)}",
439
  f"--train_batch_size={int(train_batch_size)}",
440
  f"--repeats={int(repeats)}",
441
  f"--gradient_accumulation_steps={int(gradient_accumulation_steps)}",
 
209
  text_encoder_learning_rate,
210
  seed,
211
  resolution,
212
+ center_crop,
213
  num_train_epochs,
214
  checkpointing_steps,
215
  prior_loss_weight,
 
253
  f"output_dir={slugged_lora_name}",
254
  f"mixed_precision={mixed_precision}",
255
  f"resolution={int(resolution)}",
256
+ "center_crop",
257
  f"train_batch_size={int(train_batch_size)}",
258
  f"repeats={int(repeats)}",
259
  f"gradient_accumulation_steps={int(gradient_accumulation_steps)}",
 
436
  f"--output_dir={slugged_lora_name}",
437
  f"--mixed_precision={mixed_precision}",
438
  f"--resolution={int(resolution)}",
439
+ "--center_crop",
440
  f"--train_batch_size={int(train_batch_size)}",
441
  f"--repeats={int(repeats)}",
442
  f"--gradient_accumulation_steps={int(gradient_accumulation_steps)}",