Spaces:
Sleeping
Sleeping
Murali Chandran commited on
Commit ·
a676506
1
Parent(s): 444ed4b
fixed inference
Browse files- bot_inference.py +2 -1
bot_inference.py
CHANGED
|
@@ -5,7 +5,8 @@ from bot_model import GPT, decode, greedy_until_eos, extract_numbers, patch_fina
|
|
| 5 |
# Load checkpoint
|
| 6 |
checkpoint = hf_hub_download(
|
| 7 |
repo_id="codeninja3d/minigpt-model",
|
| 8 |
-
filename="model_latest.pt"
|
|
|
|
| 9 |
)
|
| 10 |
|
| 11 |
stoi, itos = checkpoint["vocab"]
|
|
|
|
| 5 |
# Load checkpoint
|
| 6 |
checkpoint = hf_hub_download(
|
| 7 |
repo_id="codeninja3d/minigpt-model",
|
| 8 |
+
filename="model_latest.pt",
|
| 9 |
+
repo_type="model"
|
| 10 |
)
|
| 11 |
|
| 12 |
stoi, itos = checkpoint["vocab"]
|