diff --git a/tests/test_image_onnx_embeddings.py b/tests/test_image_onnx_embeddings.py index 78194caf..a5fb8e36 100644 --- a/tests/test_image_onnx_embeddings.py +++ b/tests/test_image_onnx_embeddings.py @@ -21,6 +21,9 @@ "Qdrant/Unicom-ViT-B-32": np.array( [0.0418, 0.0550, 0.0003, 0.0253, -0.0185, 0.0016, -0.0368, -0.0402, -0.0891, -0.0186] ), + "jinaai/jina-clip-v1": np.array( + [-0.029, 0.0216, 0.0396, 0.0283, -0.0023, 0.0151, 0.011, -0.0235, 0.0251, -0.0343] + ), } diff --git a/tests/test_text_onnx_embeddings.py b/tests/test_text_onnx_embeddings.py index 328d2831..36bf82c1 100644 --- a/tests/test_text_onnx_embeddings.py +++ b/tests/test_text_onnx_embeddings.py @@ -75,7 +75,7 @@ def test_embedding(): dim = model_desc["dim"] - model = TextEmbedding(model_name=model_desc["model"], cache_dir="models") + model = TextEmbedding(model_name=model_desc["model"]) docs = ["hello world", "flag embedding"] embeddings = list(model.embed(docs)) embeddings = np.stack(embeddings, axis=0)