Skip to content

Commit 55b1b47

Browse files
committed
updated num workers
1 parent fb06b56 commit 55b1b47

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/litdata/utilities/hf_dataset.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def index_hf_dataset(hf_url: str) -> str:
2020
print(f"Found HF index.json file in {cache_index_path}.")
2121
else:
2222
print("Indexing HF dataset...")
23-
index_parquet_dataset(hf_url, cache_dir, remove_after_indexing=False)
23+
index_parquet_dataset(hf_url, cache_dir, remove_after_indexing=False, num_workers=os.cpu_count())
2424

2525
print("=" * 50)
2626

0 commit comments

Comments
 (0)