From 0fedc392d57aa637a285f56683964bf0b426a309 Mon Sep 17 00:00:00 2001
From: Ashish Kumar Singh <ashishkmr472@gmail.com>
Date: Fri, 2 Aug 2024 13:05:15 +0000
Subject: [PATCH] yet more

---
 training.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/training.py b/training.py
index f0a7db1..619d5e3 100644
--- 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: