yet more

AshishKumar4 · Aug 2, 2024 · 0fedc39 · 0fedc39
1 parent 644f4bc
commit 0fedc39
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/training.py b/training.py
@@ -927,6 +927,7 @@ def main(args):
         grain_read_buffer_size=GRAIN_READ_BUFFER_SIZE, grain_worker_buffer_size=GRAIN_WORKER_BUFFER_SIZE,
         text_encoders=text_encoders,
         seed=args.dataset_seed,
+        dataset_source=args.dataset_path,
     )
 
     if args.dataset_test: