Commit ·
133d507
1
Parent(s): 2152c36
model_loader: point REPO at ericaRC/example and include TROUBLESHOOTING.md
Browse filesPrevious value (Resilient-Coders/baseline-nllb) was the pre-migration
target and would 403 for anyone re-running the script. Also adds
TROUBLESHOOTING.md to ENDPOINT_FILES so it gets pushed by
push_endpoint_files(), and refreshes the login hint to `hf auth login`
since `huggingface-cli` is deprecated.
Made-with: Cursor
- model_loader.py +3 -3
model_loader.py
CHANGED
|
@@ -5,7 +5,7 @@ Pushes both the weights/tokenizer and the Inference-Endpoints artifacts
|
|
| 5 |
"Deploy → Inference Endpoints" click on the Hub just works.
|
| 6 |
|
| 7 |
Usage:
|
| 8 |
-
|
| 9 |
python model_loader.py
|
| 10 |
"""
|
| 11 |
|
|
@@ -17,9 +17,9 @@ from huggingface_hub import HfApi
|
|
| 17 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
| 18 |
|
| 19 |
BASE = "facebook/nllb-200-distilled-600M"
|
| 20 |
-
REPO = "
|
| 21 |
|
| 22 |
-
ENDPOINT_FILES = ("handler.py", "requirements.txt", "README.md")
|
| 23 |
|
| 24 |
|
| 25 |
def push_weights() -> None:
|
|
|
|
| 5 |
"Deploy → Inference Endpoints" click on the Hub just works.
|
| 6 |
|
| 7 |
Usage:
|
| 8 |
+
hf auth login # or set HF_TOKEN
|
| 9 |
python model_loader.py
|
| 10 |
"""
|
| 11 |
|
|
|
|
| 17 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
| 18 |
|
| 19 |
BASE = "facebook/nllb-200-distilled-600M"
|
| 20 |
+
REPO = "ericaRC/example"
|
| 21 |
|
| 22 |
+
ENDPOINT_FILES = ("handler.py", "requirements.txt", "README.md", "TROUBLESHOOTING.md")
|
| 23 |
|
| 24 |
|
| 25 |
def push_weights() -> None:
|