fix: map to cpu
Browse files
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
|