From 504cf9e49102691b2f6be3941722a8ca1d65ef94 Mon Sep 17 00:00:00 2001 From: Jong Wook Kim Date: Sun, 18 Jul 2021 20:28:30 -0700 Subject: [PATCH] test fix --- tests/test_consistency.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_consistency.py b/tests/test_consistency.py index 29d343d..f2c6fd4 100644 --- a/tests/test_consistency.py +++ b/tests/test_consistency.py @@ -9,7 +9,7 @@ import clip @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)