Spaces:
Runtime error
Runtime error
Update opp.py
Browse files
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
|