yuchenlin commited on
Commit
e44fa8d
1 Parent(s): ae6f0d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ from threading import Thread
5
  from typing import Iterator
6
 
7
  # Load model and tokenizer
8
- model_name = "Magpie-Align/MagpieLM-4B-Chat-v0.1"
9
 
10
  device = "cuda" # the device to load the model onto
11
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
@@ -65,7 +65,7 @@ demo = gr.ChatInterface(
65
  respond,
66
  additional_inputs=[
67
  gr.Textbox(value="You are Magpie, a helpful AI assistant.", label="System message"),
68
- gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
69
  gr.Slider(minimum=0.1, maximum=4.0, value=0.6, step=0.1, label="Temperature"),
70
  gr.Slider(
71
  minimum=0.1,
 
5
  from typing import Iterator
6
 
7
  # Load model and tokenizer
8
+ model_name = "Magpie-Align/MagpieLM-8B-Chat-v0.1"
9
 
10
  device = "cuda" # the device to load the model onto
11
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 
65
  respond,
66
  additional_inputs=[
67
  gr.Textbox(value="You are Magpie, a helpful AI assistant.", label="System message"),
68
+ gr.Slider(minimum=1, maximum=2048, value=1024, step=1, label="Max new tokens"),
69
  gr.Slider(minimum=0.1, maximum=4.0, value=0.6, step=0.1, label="Temperature"),
70
  gr.Slider(
71
  minimum=0.1,