zhaozitian commited on
Commit
a1fbbe4
·
1 Parent(s): 2758ae4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ 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("daryl149/llama-2-7b-chat-hf")
12
 
13
  BASE_MODEL = "daryl149/llama-2-13b-chat-hf"
14
  LORA_WEIGHTS = "Sparticle/llama-2-13b-chat-japanese-lora"
@@ -138,7 +138,7 @@ g = gr.Interface(
138
  label="Output",
139
  )
140
  ],
141
- title="Llama2_7b_chat_Japanese_Lora",
142
  description="Llama-2-13b-chat-Japanese-LoRA is a multi-purpose large language model for Japanese text.\n\
143
  This model is presented by the joint effort of Sparticle Inc. and A. I. Hakusan Inc.\n\
144
  Llama-2-13b-chat-Japanese-LoRAは日本語テキストのための多目的大規模言語モデルです。\n\
 
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-13b-chat-hf")
12
 
13
  BASE_MODEL = "daryl149/llama-2-13b-chat-hf"
14
  LORA_WEIGHTS = "Sparticle/llama-2-13b-chat-japanese-lora"
 
138
  label="Output",
139
  )
140
  ],
141
+ title="Llama2_13b_chat_Japanese_Lora",
142
  description="Llama-2-13b-chat-Japanese-LoRA is a multi-purpose large language model for Japanese text.\n\
143
  This model is presented by the joint effort of Sparticle Inc. and A. I. Hakusan Inc.\n\
144
  Llama-2-13b-chat-Japanese-LoRAは日本語テキストのための多目的大規模言語モデルです。\n\