fix: test

jina-ai · Dec 9, 2022 · 78fa087 · 78fa087
1 parent 0990367
commit 78fa087
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_model.py b/tests/test_model.py
@@ -37,4 +37,6 @@ def test_onnx_model(name):
     ['ViT-H-14::laion2b-s32b-b79k'],
 )
 def test_large_onnx_model_fp16(name):
-    CLIPOnnxModel(name, dtype='fp16')
+    from clip_server.executors.clip_onnx import CLIPEncoder
+
+    CLIPEncoder(name, dtype='fp16')