Update embedding.py

xusenlinzy · Apr 27, 2024 · a70de4b · a70de4b
1 parent 6e60de5
commit a70de4b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/rag/models/embedding.py b/api/rag/models/embedding.py
@@ -64,7 +64,7 @@ def embed(
                 vecs = np.c_[vecs, zeros]
             elif 0 < dimensions < dim:
                 vecs = vecs[..., :dimensions]  # Shrink the embedding dimensions
-                vecs = normalize_embeddings(vecs).numpy()
+                vecs = normalize_embeddings(vecs).cpu().numpy()
 
             if encoding_format == "base64":
                 vecs = [base64.b64encode(v.tobytes()).decode("utf-8") for v in vecs]