Spaces:
Running
on
Zero
Running
on
Zero
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)
|