alfredplpl commited on
Commit
d97b4d3
1 Parent(s): 06a30bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -10,8 +10,8 @@ from threading import Thread
10
 
11
  DESCRIPTION = '''
12
  <div>
13
- <h1 style="text-align: center;">日本語向け Llama 3 8B</h1>
14
- <p>日本語向け Llama 3 のデモだよ。 <a href="https://huggingface.co/alfredplpl/Llama-3-8B-Instruct-Ja"><b>alfredplpl/Llama-3-8B-Instruct-Ja</b></a>.</p>
15
  </div>
16
  '''
17
 
@@ -45,8 +45,8 @@ h1 {
45
  """
46
 
47
  # Load the tokenizer and model
48
- tokenizer = AutoTokenizer.from_pretrained("alfredplpl/Llama-3-8B-Instruct-Ja")
49
- model = AutoModelForCausalLM.from_pretrained("alfredplpl/Llama-3-8B-Instruct-Ja", device_map="auto")
50
  terminators = [
51
  tokenizer.eos_token_id,
52
  tokenizer.convert_tokens_to_ids("<|eot_id|>")
 
10
 
11
  DESCRIPTION = '''
12
  <div>
13
+ <h1 style="text-align: center;">Llama-3-ELYZA-JP-8B</h1>
14
+ <p>elyza/Llama-3-ELYZA-JP-8Bの非公式デモだよ。 <a href="https://huggingface.co/elyza/Llama-3-ELYZA-JP-8B"><b>elyza/Llama-3-ELYZA-JP-8B</b></a>.</p>
15
  </div>
16
  '''
17
 
 
45
  """
46
 
47
  # Load the tokenizer and model
48
+ tokenizer = AutoTokenizer.from_pretrained("elyza/Llama-3-ELYZA-JP-8B")
49
+ model = AutoModelForCausalLM.from_pretrained("elyza/Llama-3-ELYZA-JP-8B", device_map="auto")
50
  terminators = [
51
  tokenizer.eos_token_id,
52
  tokenizer.convert_tokens_to_ids("<|eot_id|>")