amosyou commited on
Commit
a4e01fa
·
1 Parent(s): 330e8a2

fix: map to cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ model = get_model(args, ingr_vocab_size, instrs_vocab_size)
60
  # hf_hub_download(REPO_ID, 'data/modelbest.ckpt', token=HF_TOKEN), map_location=map_loc)
61
  # )
62
  model.load_state_dict(torch.load(
63
- hf_hub_download(REPO_ID, 'data/modelbest.ckpt', token=HF_TOKEN))
64
  )
65
  model.eval()
66
  model.ingrs_only = False
 
60
  # hf_hub_download(REPO_ID, 'data/modelbest.ckpt', token=HF_TOKEN), map_location=map_loc)
61
  # )
62
  model.load_state_dict(torch.load(
63
+ hf_hub_download(REPO_ID, 'data/modelbest.ckpt', token=HF_TOKEN), map_location=torch.device('cpu'))
64
  )
65
  model.eval()
66
  model.ingrs_only = False