Spaces:
Runtime error
Runtime error
fix for crop_center variable (#4)
Browse files- fix for crop_center variable (9a6b469099bb500a69e49a82803a887068f6d38e)
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 |
-
|
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 |
-
|
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)}",
|