Merge d73f2db22c into a5a162044c

2026-04-17 09:18:00 +00:00 · 2025-11-03 14:58:48 +01:00
parent a5a162044c d73f2db22c
commit cfff446aab
18 changed files with 242 additions and 2 deletions
--- a/sd3_train.py
+++ b/sd3_train.py
@@ -502,6 +502,7 @@ def train(args):
        shuffle=True,
        collate_fn=collator,
        num_workers=n_workers,
+        pin_memory=args.pin_memory,
        persistent_workers=args.persistent_data_loader_workers,
    )