Spaces:
Paused
Paused
Tao Wu
commited on
Commit
·
50220da
1
Parent(s):
f6aa47a
fix bug
Browse files- 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("
|
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")
|