test fix
This commit is contained in:
parent
a2737ac264
commit
504cf9e491
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue