Spaces:
Runtime error
Runtime error
increase max sample size, center_crop = True
#2
by
dkebudi
- opened
app.py
CHANGED
|
@@ -18,7 +18,7 @@ from pathlib import Path
|
|
| 18 |
import spaces
|
| 19 |
import zipfile
|
| 20 |
|
| 21 |
-
MAX_IMAGES =
|
| 22 |
|
| 23 |
is_spaces = True if os.environ.get('SPACE_ID') else False
|
| 24 |
|
|
@@ -246,6 +246,7 @@ def start_training(
|
|
| 246 |
commands = [
|
| 247 |
"pretrained_model_name_or_path=stabilityai/stable-diffusion-xl-base-1.0",
|
| 248 |
"pretrained_vae_model_name_or_path=madebyollin/sdxl-vae-fp16-fix",
|
|
|
|
| 249 |
f"instance_prompt={concept_sentence}",
|
| 250 |
f"dataset_name=./{dataset_folder}",
|
| 251 |
"caption_column=prompt",
|
|
|
|
| 18 |
import spaces
|
| 19 |
import zipfile
|
| 20 |
|
| 21 |
+
MAX_IMAGES = 500
|
| 22 |
|
| 23 |
is_spaces = True if os.environ.get('SPACE_ID') else False
|
| 24 |
|
|
|
|
| 246 |
commands = [
|
| 247 |
"pretrained_model_name_or_path=stabilityai/stable-diffusion-xl-base-1.0",
|
| 248 |
"pretrained_vae_model_name_or_path=madebyollin/sdxl-vae-fp16-fix",
|
| 249 |
+
f"center_crop={bool(True)}",
|
| 250 |
f"instance_prompt={concept_sentence}",
|
| 251 |
f"dataset_name=./{dataset_folder}",
|
| 252 |
"caption_column=prompt",
|