iatechaways commited on
Commit
32e01fc
·
1 Parent(s): dcf17b8

Update BASE_MODEL in model_loader.py to correct model path

Browse files
Files changed (1) hide show
  1. 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/Meta-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
 
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