test embedder

StampyAI · Oct 16, 2023 · 0f97d54 · 0f97d54
1 parent 57db939
commit 0f97d54
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/align_data/embeddings/embedding_utils.py b/align_data/embeddings/embedding_utils.py
@@ -105,6 +105,7 @@ def moderation_check(texts: List[str], max_texts_num: int = 32) -> List[Moderati
 @handle_openai_errors
 def _single_batch_compute_openai_embeddings(batch: List[str], **kwargs) -> List[List[float]]:
     """Compute embeddings for a batch."""
+    print('emebedding', batch[0][:10], 'of len', len(batch))
     batch_data = openai.Embedding.create(input=batch, engine=OPENAI_EMBEDDINGS_MODEL, **kwargs).data
     return [d["embedding"] for d in batch_data]