JosephusCheung commited on
Commit
9b70592
Β·
verified Β·
1 Parent(s): 850186a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -55,7 +55,7 @@ def stream_chat(message: str, history: list, temperature: float, max_length: int
55
  do_sample=True,
56
  top_k=1,
57
  temperature=temperature,
58
- repetition_penalty=1.2,
59
  )
60
  gen_kwargs = {**input_ids, **generate_kwargs}
61
 
@@ -92,9 +92,9 @@ with gr.Blocks(css=CSS) as demo:
92
  ),
93
  gr.Slider(
94
  minimum=128,
95
- maximum=8192,
96
  step=1,
97
- value=1024,
98
  label="Max Length",
99
  render=False,
100
  ),
 
55
  do_sample=True,
56
  top_k=1,
57
  temperature=temperature,
58
+ repetition_penalty=1,
59
  )
60
  gen_kwargs = {**input_ids, **generate_kwargs}
61
 
 
92
  ),
93
  gr.Slider(
94
  minimum=128,
95
+ maximum=64000,
96
  step=1,
97
+ value=64000,
98
  label="Max Length",
99
  render=False,
100
  ),