seedmanc commited on
Commit
ba32bfd
1 Parent(s): 3271ef3

Update opp.py

Browse files
Files changed (1) hide show
  1. opp.py +1 -1
opp.py CHANGED
@@ -71,7 +71,7 @@ def load_models():
71
  model.to(device)
72
  model.eval()
73
 
74
- model2, preprocess = clip.load(checkpoint, device=device)
75
 
76
  model_dict = {}
77
  model_dict['classifier'] = model
 
71
  model.to(device)
72
  model.eval()
73
 
74
+ model2, preprocess = clip.load(checkpoint, device=device, jit=True)
75
 
76
  model_dict = {}
77
  model_dict['classifier'] = model