increase max sample size, center_crop = True

#2
by dkebudi - opened
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,7 +18,7 @@ from pathlib import Path
18
  import spaces
19
  import zipfile
20
 
21
- MAX_IMAGES = 150
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",