fix: embedding 与 model_name 类型检查

2024-01-15 14:21:01 +08:00
parent dabdbb42de
commit 70a08a9ed1
2 changed files with 5 additions and 3 deletions
--- a/acge_embedding.py
+++ b/acge_embedding.py
@@ -48,4 +48,4 @@ def acge_embedding(text: list[str]) -> list[list[float]]:
        norm="l2",
        axis=1,
    )
-    return vector
+    return vector.tolist()