dwb2023 commited on
Commit
4a8458f
1 Parent(s): f369df4

Update app.py

Browse files

change tokenizer to one provided directly from meta: meta-llama/Llama-2-7b-hf

Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -42,7 +42,8 @@ model = AutoModelForCausalLM.from_pretrained(
42
  trust_remote_code=True,
43
  )
44
  tokenizer = AutoTokenizer.from_pretrained(
45
- "NousResearch/Llama-2-7b-hf",
 
46
  trust_remote_code=True,
47
  tokenizer_class=LlamaTokenizer,
48
  )
 
42
  trust_remote_code=True,
43
  )
44
  tokenizer = AutoTokenizer.from_pretrained(
45
+ # "NousResearch/Llama-2-7b-hf",
46
+ "meta-llama/Llama-2-7b-hf",
47
  trust_remote_code=True,
48
  tokenizer_class=LlamaTokenizer,
49
  )