Spaces:
Paused
Paused
Commit ·
32e01fc
1
Parent(s): dcf17b8
Update BASE_MODEL in model_loader.py to correct model path
Browse files- model_loader.py +1 -1
model_loader.py
CHANGED
|
@@ -8,7 +8,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
|
| 8 |
from peft import PeftModel
|
| 9 |
|
| 10 |
# CONFIG - change these if your repo name differs
|
| 11 |
-
BASE_MODEL = "meta-llama/
|
| 12 |
ADAPTER = "iatechaways/crm_model" # your LoRA adapter on HF
|
| 13 |
|
| 14 |
# Get HF token from environment
|
|
|
|
| 8 |
from peft import PeftModel
|
| 9 |
|
| 10 |
# CONFIG - change these if your repo name differs
|
| 11 |
+
BASE_MODEL = "meta-llama/Llama-3.2-3B" # or local path if you prefer
|
| 12 |
ADAPTER = "iatechaways/crm_model" # your LoRA adapter on HF
|
| 13 |
|
| 14 |
# Get HF token from environment
|