Tao Wu commited on
Commit
50220da
·
1 Parent(s): f6aa47a
Files changed (1) hide show
  1. app/embedding_setup.py +1 -1
app/embedding_setup.py CHANGED
@@ -33,7 +33,7 @@ retriever = db.as_retriever(search_kwargs={"k": TOP_K})
33
  LLM_MODEL = "meta-llama/Meta-Llama-3-8B-Instruct"
34
  lora_weights = "/hpcwork/vg380347/llama3/Instruct_8B_EngGer_alpaca_finetune_pairwise_skill_24_128/last_checkpoint"
35
 
36
- hf_auth = os.environ.get("SECRET_EXAMPLE")
37
 
38
 
39
  tokenizer = AutoTokenizer.from_pretrained(LLM_MODEL, token=hf_auth, cache_dir="/hpcwork/vg380347/.cache")
 
33
  LLM_MODEL = "meta-llama/Meta-Llama-3-8B-Instruct"
34
  lora_weights = "/hpcwork/vg380347/llama3/Instruct_8B_EngGer_alpaca_finetune_pairwise_skill_24_128/last_checkpoint"
35
 
36
+ hf_auth = os.environ.get("hf_token")
37
 
38
 
39
  tokenizer = AutoTokenizer.from_pretrained(LLM_MODEL, token=hf_auth, cache_dir="/hpcwork/vg380347/.cache")