tmartinez commited on
Commit
d654128
1 Parent(s): 695fcc9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -9,7 +9,9 @@ from huggingface_hub import login
9
  login('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')
10
 
11
  # Model name in Hugging Face docs
12
- model = "meta-llama/Llama-2-7b-chat-hf"
 
 
13
 
14
  tokenizer = AutoTokenizer.from_pretrained(model, use_auth_token=True)
15
 
 
9
  login('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')
10
 
11
  # Model name in Hugging Face docs
12
+ model = 'klyang/MentaLLaMA-chat-7B'
13
+
14
+ #"meta-llama/Llama-2-7b-chat-hf"
15
 
16
  tokenizer = AutoTokenizer.from_pretrained(model, use_auth_token=True)
17