Spaces:
Paused
Paused
Update lavila/models/openai_clip.py
Browse files
lavila/models/openai_clip.py
CHANGED
@@ -121,7 +121,7 @@ def load(name: str, device: Union[str, torch.device] = "cuda" if torch.cuda.is_a
|
|
121 |
A torchvision transform that converts a PIL image into a tensor that the returned model can take as its input
|
122 |
"""
|
123 |
if name in _MODELS:
|
124 |
-
model_path = _download(_MODELS[name], download_root or os.path.expanduser("
|
125 |
elif os.path.isfile(name):
|
126 |
model_path = name
|
127 |
else:
|
|
|
121 |
A torchvision transform that converts a PIL image into a tensor that the returned model can take as its input
|
122 |
"""
|
123 |
if name in _MODELS:
|
124 |
+
model_path = _download(_MODELS[name], download_root or os.path.expanduser("cache/clip"))
|
125 |
elif os.path.isfile(name):
|
126 |
model_path = name
|
127 |
else:
|