Spaces:
Sleeping
Sleeping
token
Browse files- app/model_loader.py +2 -2
app/model_loader.py
CHANGED
|
@@ -4,9 +4,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
| 4 |
from peft import PeftModel
|
| 5 |
|
| 6 |
def load_model():
|
| 7 |
-
hf_token = os.getenv("
|
| 8 |
if not hf_token:
|
| 9 |
-
raise RuntimeError("
|
| 10 |
|
| 11 |
# Use a user-writable cache directory (important for Docker non-root)
|
| 12 |
HF_CACHE = os.path.expanduser("~/.cache/huggingface")
|
|
|
|
| 4 |
from peft import PeftModel
|
| 5 |
|
| 6 |
def load_model():
|
| 7 |
+
hf_token = os.getenv("hf_token")
|
| 8 |
if not hf_token:
|
| 9 |
+
raise RuntimeError("hf_token not set.")
|
| 10 |
|
| 11 |
# Use a user-writable cache directory (important for Docker non-root)
|
| 12 |
HF_CACHE = os.path.expanduser("~/.cache/huggingface")
|