Spaces:
Runtime error
Runtime error
fix for crop_center variable
#4
by
dkebudi
- opened
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)}",
|