Spaces:

isl-org
/

lang-seg

Runtime error

akhaliq HF Staff commited on Apr 11, 2022

Commit

f347ac1

1 Parent(s): ea40b51

Update modules/models/lseg_vit.py

Files changed (1) hide show

modules/models/lseg_vit.py CHANGED Viewed

@@ -221,7 +221,7 @@ def get_readout_oper(vit_features, features, use_readout, start_index=1):
 def _make_pretrained_clip_vitl16_384(
     pretrained, use_readout="ignore", hooks=None, enable_attention_hooks=False
 ):
-    clip_pretrained, _ = clip.load("ViT-B/32", device='cuda', jit=False)
     model = timm.create_model("vit_large_patch16_384", pretrained=pretrained)
     hooks = [5, 11, 17, 23] if hooks == None else hooks

 def _make_pretrained_clip_vitl16_384(
     pretrained, use_readout="ignore", hooks=None, enable_attention_hooks=False
 ):
+    clip_pretrained, _ = clip.load("ViT-B/32", device='cpu', jit=False)
     model = timm.create_model("vit_large_patch16_384", pretrained=pretrained)
     hooks = [5, 11, 17, 23] if hooks == None else hooks