Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|
136 |
-
gr.Slider(
|
137 |
-
gr.Slider(
|
138 |
-
gr.Slider(
|
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 |
|