devve1 commited on
Commit
7e7d581
1 Parent(s): fa422ef

Update cluster_semantic_chunker.py

Browse files
Files changed (1) hide show
  1. cluster_semantic_chunker.py +1 -1
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, batch_size=1)
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)