localhorst/CLIP_on_Tesla_K20Xm
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

test fix

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_consistency.py
View File

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