Spaces:
Running
Running
jbdel
commited on
Commit
•
8f1a208
1
Parent(s):
c095dce
push with token
Browse files- paper_chat_tab.py +2 -1
paper_chat_tab.py
CHANGED
@@ -166,7 +166,8 @@ def create_chat_interface(model_name, paper_content, hf_token_input):
|
|
166 |
if model_name not in tokenizer_cache:
|
167 |
# Load the tokenizer from Hugging Face
|
168 |
# tokenizer = AutoTokenizer.from_pretrained(model_name)
|
169 |
-
tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B-Instruct"
|
|
|
170 |
tokenizer_cache[model_name] = tokenizer
|
171 |
else:
|
172 |
tokenizer = tokenizer_cache[model_name]
|
|
|
166 |
if model_name not in tokenizer_cache:
|
167 |
# Load the tokenizer from Hugging Face
|
168 |
# tokenizer = AutoTokenizer.from_pretrained(model_name)
|
169 |
+
tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B-Instruct",
|
170 |
+
token=os.environ.get("HF_TOKEN"))
|
171 |
tokenizer_cache[model_name] = tokenizer
|
172 |
else:
|
173 |
tokenizer = tokenizer_cache[model_name]
|