Hugging Face's logo

Spaces:
tennant
/
MUG_caption
Runtime error

App Files Community
tennant commited on
Commit
b14c338
·
1 Parent(s): a19b5b3

add location

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from huggingface_hub import hf_hub_download
10
 
11
  tokenizer = AutoTokenizer.from_pretrained('bert-base-uncased', )
12
 
13
- ckpt = torch.load(hf_hub_download('tennant/MUG', 'mae_bert_vit_b_cc3m.pth'))
14
 
15
  new_dict = OrderedDict()
16
  for k, v in ckpt.items():
 
10
 
11
  tokenizer = AutoTokenizer.from_pretrained('bert-base-uncased', )
12
 
13
+ ckpt = torch.load(hf_hub_download('tennant/MUG', 'mae_bert_vit_b_cc3m.pth'), map_location='cpu')
14
 
15
  new_dict = OrderedDict()
16
  for k, v in ckpt.items():