Update app.py
Browse files
app.py
CHANGED
|
@@ -5,6 +5,7 @@ from huggingface_hub import login
|
|
| 5 |
import torch
|
| 6 |
import os
|
| 7 |
hf_token = os.getenv("llama")
|
|
|
|
| 8 |
#login(hf_token)
|
| 9 |
# Model and adapter paths
|
| 10 |
model_name = "unsloth/llama-3.2-1b-instruct-bnb-4bit" # Base model
|
|
|
|
| 5 |
import torch
|
| 6 |
import os
|
| 7 |
hf_token = os.getenv("llama")
|
| 8 |
+
assert hf_token is not None, "Token is missing! Make sure 'llama' is set in the environment."
|
| 9 |
#login(hf_token)
|
| 10 |
# Model and adapter paths
|
| 11 |
model_name = "unsloth/llama-3.2-1b-instruct-bnb-4bit" # Base model
|