Hugging Face's logo

Spaces:
czq
/
TimeQA_demo
Sleeping

App Files Community
czq commited on
Commit
14f7217
·
1 Parent(s): 77292c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -62,7 +62,7 @@ model = BigBirdForQuestionAnsweringWithNull(config, model_id)
62
  # model.to('cuda')
63
  model.eval()
64
 
65
- model.load_state_dict(torch.load(model_path+'/pytorch_model.bin')) # map_location是指定加载到哪个设备
66
 
67
  tokenizer = BigBirdTokenizer.from_pretrained(model_path)
68
 
 
62
  # model.to('cuda')
63
  model.eval()
64
 
65
+ model.load_state_dict(torch.load(model_path+'/pytorch_model.bin',map_location=torch.device('cpu'))) # map_location是指定加载到哪个设备
66
 
67
  tokenizer = BigBirdTokenizer.from_pretrained(model_path)
68