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())
|
@pytest.mark.parametrize('model_name', clip.available_models())
|
||||||
def test_consistency(model_name):
|
def test_consistency(model_name):
|
||||||
device = "cpu"
|
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)
|
py_model, _ = clip.load(model_name, device=device, jit=False)
|
||||||
|
|
||||||
image = transform(Image.open("CLIP.png")).unsqueeze(0).to(device)
|
image = transform(Image.open("CLIP.png")).unsqueeze(0).to(device)
|
||||||
|
|
Loading…
Reference in New Issue