eliot-hub commited on
Commit
8216547
·
1 Parent(s): 40e77db
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ model_emb = "mixedbread-ai/mxbai-embed-large-v1"
41
 
42
  # Set up ChromaDB
43
  memoires_ds = load_dataset("eliot-hub/memoires_vec_800", split="data", token=HF_TOKEN, streaming=True)
44
- batched_ds = memoires_ds.batch(batch_size=50000)
45
  client = chromadb.Client()
46
  collection = client.get_or_create_collection(name="embeddings_mxbai")
47
 
 
41
 
42
  # Set up ChromaDB
43
  memoires_ds = load_dataset("eliot-hub/memoires_vec_800", split="data", token=HF_TOKEN, streaming=True)
44
+ batched_ds = memoires_ds.batch(batch_size=41000)
45
  client = chromadb.Client()
46
  collection = client.get_or_create_collection(name="embeddings_mxbai")
47