We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent fb06b56 commit 55b1b47Copy full SHA for 55b1b47
src/litdata/utilities/hf_dataset.py
@@ -20,7 +20,7 @@ def index_hf_dataset(hf_url: str) -> str:
20
print(f"Found HF index.json file in {cache_index_path}.")
21
else:
22
print("Indexing HF dataset...")
23
- index_parquet_dataset(hf_url, cache_dir, remove_after_indexing=False)
+ index_parquet_dataset(hf_url, cache_dir, remove_after_indexing=False, num_workers=os.cpu_count())
24
25
print("=" * 50)
26
0 commit comments