updated defaults
Browse files
app.py
CHANGED
@@ -67,8 +67,8 @@ def main ():
|
|
67 |
gr.Textbox(lines=20), # prompt
|
68 |
gr.Dropdown(["GPT-JT-6B-v1"]), # model_name
|
69 |
gr.Slider(1, 256, value=200), # max_tokens
|
70 |
-
gr.Slider(0.0, 0
|
71 |
-
gr.Slider(0.0,
|
72 |
gr.Slider(0, 1000, value=40) # top_k
|
73 |
],
|
74 |
outputs=gr.Textbox(lines=7)
|
|
|
67 |
gr.Textbox(lines=20), # prompt
|
68 |
gr.Dropdown(["GPT-JT-6B-v1"]), # model_name
|
69 |
gr.Slider(1, 256, value=200), # max_tokens
|
70 |
+
gr.Slider(0.0, 10.0, value=0.1), # temperature
|
71 |
+
gr.Slider(0.0, 1.0, value=0.9), # top_p
|
72 |
gr.Slider(0, 1000, value=40) # top_k
|
73 |
],
|
74 |
outputs=gr.Textbox(lines=7)
|