No shuffle option for sd3

sd3_train.py

@@ -504,7 +504,7 @@ def train(args):
     train_dataloader = torch.utils.data.DataLoader(
         train_dataset_group,
         batch_size=1,
-        shuffle=True,
+        shuffle=not args.no_shuffle,
         collate_fn=collator,
         num_workers=n_workers,
         persistent_workers=args.persistent_data_loader_workers,
@@ -1079,6 +1079,11 @@ def setup_parser() -> argparse.ArgumentParser:
         action="append",
         nargs=2
     )
+    parser.add_argument(
+        "--no_shuffle",
+        action="store_true",
+        help="do not use shuffle in the dataloader - rely on the underlying dataset",
+    )
     return parser