Instructions to use Clementio/PLRS with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Clementio/PLRS with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("Clementio/PLRS", dtype="auto") - Notebooks
- Google Colab
- Kaggle
Clementina Tom (via Gemini) commited on
Commit ·
e5cd6dd
1
Parent(s): a30026f
Fix model paths: looking in models/ folder
Browse files
plrs/model/model_loader.py
CHANGED
|
@@ -41,9 +41,9 @@ def load_model_from_hub(device: str = "cpu"):
|
|
| 41 |
return None, "huggingface_hub not installed"
|
| 42 |
|
| 43 |
for filename, model_type in [
|
| 44 |
-
("sakt_decay_best.pt", "SAKTWithDecay"),
|
| 45 |
-
("sakt_vanilla_best.pt", "SAKTModel"),
|
| 46 |
-
("sakt_model.pt", "SAKTModel"),
|
| 47 |
]:
|
| 48 |
try:
|
| 49 |
path = hf_hub_download(repo_id=HF_REPO, filename=filename)
|
|
|
|
| 41 |
return None, "huggingface_hub not installed"
|
| 42 |
|
| 43 |
for filename, model_type in [
|
| 44 |
+
("models/sakt_decay_best.pt", "SAKTWithDecay"),
|
| 45 |
+
("models/sakt_vanilla_best.pt", "SAKTModel"),
|
| 46 |
+
("models/sakt_model.pt", "SAKTModel"),
|
| 47 |
]:
|
| 48 |
try:
|
| 49 |
path = hf_hub_download(repo_id=HF_REPO, filename=filename)
|