sergey21000 commited on
Commit
ced57cc
·
verified ·
1 Parent(s): ef59540

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -130,12 +130,11 @@ def get_system_prompt_component(interactive: bool) -> gr.Textbox:
130
 
131
 
132
  def get_generate_args(do_sample: bool) -> List[gr.component]:
133
- visible = do_sample
134
  generate_args = [
135
- gr.Slider(label='temperature', value=GENERATE_KWARGS['temperature'], minimum=0.1, maximum=3, step=0.1, visible=visible),
136
- gr.Slider(label='top_p', value=GENERATE_KWARGS['top_p'], minimum=0.1, maximum=1, step=0.1, visible=visible),
137
- gr.Slider(label='top_k', value=GENERATE_KWARGS['top_k'], minimum=1, maximum=50, step=5, visible=visible),
138
- gr.Slider(label='repeat_penalty', value=GENERATE_KWARGS['repeat_penalty'], minimum=1, maximum=5, step=0.1, visible=visible),
139
  ]
140
  return generate_args
141
 
 
130
 
131
 
132
  def get_generate_args(do_sample: bool) -> List[gr.component]:
 
133
  generate_args = [
134
+ gr.Slider(minimum=0.1, maximum=3, value=GENERATE_KWARGS['temperature'], step=0.1, label='temperature', visible=do_sample),
135
+ gr.Slider(minimum=0, maximum=1, value=GENERATE_KWARGS['top_p'], step=0.01, label='top_p', visible=do_sample),
136
+ gr.Slider(minimum=1, maximum=50, value=GENERATE_KWARGS['top_k'], step=1, label='top_k', visible=do_sample),
137
+ gr.Slider(minimum=1, maximum=5, value=GENERATE_KWARGS['repeat_penalty'], step=0.1, label='repeat_penalty', visible=do_sample),
138
  ]
139
  return generate_args
140