vilarin commited on
Commit
3279b8c
1 Parent(s): bd0158c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -55,7 +55,7 @@ def generate(message, history, system, max_tokens, temperature, top_p, top_k, pe
55
  top_k=top_k,
56
  repetition_penalty=penalty,
57
  max_tokens=max_tokens,
58
- eos_token_id=[151645,151643],
59
  )
60
  # generate outputs
61
  llm = LLM(model=model)
@@ -89,7 +89,14 @@ with gr.Blocks(css=css) as demo:
89
  label="Max tokens",
90
  render=False,
91
  ),
92
- gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
 
 
 
 
 
 
 
93
  gr.Slider(
94
  minimum=0.1,
95
  maximum=1.0,
 
55
  top_k=top_k,
56
  repetition_penalty=penalty,
57
  max_tokens=max_tokens,
58
+ stop_token_ids=[151645,151643],
59
  )
60
  # generate outputs
61
  llm = LLM(model=model)
 
89
  label="Max tokens",
90
  render=False,
91
  ),
92
+ gr.Slider(
93
+ minimum=0.1,
94
+ maximum=1.0,
95
+ value=0.7,
96
+ step=0.1,
97
+ label="Temperature",
98
+ render=False,
99
+ ),
100
  gr.Slider(
101
  minimum=0.1,
102
  maximum=1.0,