zhaozitian commited on
Commit
6357cd4
·
1 Parent(s): 1c9d59b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,10 +8,10 @@ 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("decapoda-research/llama-7b-hf")
12
 
13
- BASE_MODEL = "decapoda-research/llama-7b-hf"
14
- LORA_WEIGHTS = "tloen/alpaca-lora-7b"
15
 
16
  if torch.cuda.is_available():
17
  device = "cuda"
 
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")
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():
17
  device = "cuda"