zhaozitian commited on
Commit
6231256
1 Parent(s): 437f002

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,9 +8,9 @@ assert (
8
  ), "LLaMA is now in HuggingFace's main branch.\nPlease reinstall it: pip uninstall transformers && pip install git+https://github.com/huggingface/transformers.git"
9
  from transformers import LlamaTokenizer, LlamaForCausalLM, GenerationConfig
10
 
11
- tokenizer = LlamaTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf", use_auth_token=True)
12
 
13
- BASE_MODEL = "meta-llama/Llama-2-7b-chat-hf"
14
  LORA_WEIGHTS = "Sparticle/llama-2-7b-japanese-lora"
15
 
16
  if torch.cuda.is_available():
 
8
  ), "LLaMA is now in HuggingFace's main branch.\nPlease reinstall it: pip uninstall transformers && pip install git+https://github.com/huggingface/transformers.git"
9
  from transformers import LlamaTokenizer, LlamaForCausalLM, GenerationConfig
10
 
11
+ tokenizer = LlamaTokenizer.from_pretrained("daryl149/llama-2-7b-chat-hf", use_auth_token=True)
12
 
13
+ BASE_MODEL = "daryl149/llama-2-7b-chat-hf"
14
  LORA_WEIGHTS = "Sparticle/llama-2-7b-japanese-lora"
15
 
16
  if torch.cuda.is_available():