eliot-hub commited on
Commit
bfa3459
·
1 Parent(s): 1bea5ac

batch_size

Browse files
Files changed (1) hide show
  1. hf_to_chroma_ds.py +1 -1
hf_to_chroma_ds.py CHANGED
@@ -69,7 +69,7 @@ class Memoires_DS(Dataset):
69
 
70
 
71
 
72
- def import_into_chroma(chroma_client, dataset, collection_name=None, embedding_function=None, batch_size=30000):
73
  """
74
  Imports a dataset into Chroma in batches.
75
 
 
69
 
70
 
71
 
72
+ def import_into_chroma(chroma_client, dataset, collection_name=None, embedding_function=None, batch_size=5000):
73
  """
74
  Imports a dataset into Chroma in batches.
75