Spaces:
Paused
Paused
Update eva_clip/timm_model.py
Browse files- 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)
|