Merge pull request #168 from bghira/feature/resize-by-area

Pass resolution_type into VAECache (SDXL)
bghira · Sep 21, 2023 · 98c136d · 98c136d
2 parents 13aa1bf + 2bb34e3
commit 98c136d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train_sdxl.py b/train_sdxl.py
@@ -803,6 +803,7 @@ def collate_fn(batch):
         data_backend=data_backend,
         delete_problematic_images=args.delete_problematic_images,
         resolution=args.resolution,
+        resolution_type=args.resolution_type,
         vae_batch_size=args.vae_batch_size,
         write_batch_size=args.write_batch_size,
     )