This commit is contained in:
Jong Wook Kim 2021-07-18 20:28:30 -07:00
parent a2737ac264
commit 504cf9e491
1 changed files with 1 additions and 1 deletions

View File

@ -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)