AndaiMD commited on
Commit
c4b52ab
·
1 Parent(s): 40b2e18
Files changed (1) hide show
  1. 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("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")
 
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")