gina9726 commited on
Commit
8d48b18
·
verified ·
1 Parent(s): 1f6ef20

Update lavila/models/openai_clip.py

Browse files
Files changed (1) hide show
  1. lavila/models/openai_clip.py +1 -1
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("/store/nosnap/.cache/clip"))
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: