diff --git a/train_sdxl.py b/train_sdxl.py
index dd9e5498..826cd7ac 100644
--- 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,
     )