increase max sample size, center_crop = True

Center crop will also introduce padding if the size of the image is smaller than the resolution requests. (i.e. resolution=1920x1920 on a 1920x1080 image set)

as covered here: https://pytorch.org/vision/0.9/transforms.html

app.py

@@ -18,7 +18,7 @@ from pathlib import Path
 import spaces
 import zipfile
 
-MAX_IMAGES = 150
+MAX_IMAGES = 500
 
 is_spaces = True if os.environ.get('SPACE_ID') else False
 
@@ -246,6 +246,7 @@ def start_training(
     commands = [
         "pretrained_model_name_or_path=stabilityai/stable-diffusion-xl-base-1.0",
         "pretrained_vae_model_name_or_path=madebyollin/sdxl-vae-fp16-fix",
+        f"center_crop={bool(True)}",
         f"instance_prompt={concept_sentence}",
         f"dataset_name=./{dataset_folder}",
         "caption_column=prompt",