yanze commited on
Commit
683a2fe
1 Parent(s): 7ec536c

Update eva_clip/timm_model.py

Browse files
Files changed (1) hide show
  1. eva_clip/timm_model.py +2 -1
eva_clip/timm_model.py CHANGED
@@ -42,7 +42,8 @@ class TimmModel(nn.Module):
42
  pretrained=False):
43
  super().__init__()
44
  if timm is None:
45
- raise RuntimeError("Please `pip install timm` to use timm models.")
 
46
 
47
  self.image_size = to_2tuple(image_size)
48
  self.trunk = timm.create_model(model_name, pretrained=pretrained)
 
42
  pretrained=False):
43
  super().__init__()
44
  if timm is None:
45
+ # raise RuntimeError("Please `pip install timm` to use timm models.")
46
+ return
47
 
48
  self.image_size = to_2tuple(image_size)
49
  self.trunk = timm.create_model(model_name, pretrained=pretrained)