Update cluster_semantic_chunker.py
Browse files
cluster_semantic_chunker.py
CHANGED
@@ -27,7 +27,7 @@ class ClusterSemanticChunker(BaseChunker):
|
|
27 |
|
28 |
for i in range(0, N, BATCH_SIZE):
|
29 |
batch_sentences = sentences[i:i+BATCH_SIZE]
|
30 |
-
embeddings = embedding_function.embed_documents(batch_sentences,
|
31 |
|
32 |
# Convert embeddings list of lists to numpy array
|
33 |
batch_embedding_matrix = np.array(embeddings)
|
|
|
27 |
|
28 |
for i in range(0, N, BATCH_SIZE):
|
29 |
batch_sentences = sentences[i:i+BATCH_SIZE]
|
30 |
+
embeddings = embedding_function.embed_documents(batch_sentences, 1)
|
31 |
|
32 |
# Convert embeddings list of lists to numpy array
|
33 |
batch_embedding_matrix = np.array(embeddings)
|