test fix

SuperMousse · Jul 19, 2021 · 504cf9e · 504cf9e
1 parent a2737ac
commit 504cf9e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_consistency.py b/tests/test_consistency.py
@@ -9,7 +9,7 @@
 @pytest.mark.parametrize('model_name', clip.available_models())
 def test_consistency(model_name):
     device = "cpu"
-    jit_model, transform = clip.load(model_name, device=device)
+    jit_model, transform = clip.load(model_name, device=device, jit=True)
     py_model, _ = clip.load(model_name, device=device, jit=False)
 
     image = transform(Image.open("CLIP.png")).unsqueeze(0).to(device)