Spaces:
Sleeping
Sleeping
batchsize
Browse files
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=
|
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 |
|