Nick088 commited on
Commit
cebd695
·
verified ·
1 Parent(s): 340b3ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -9
app.py CHANGED
@@ -25,15 +25,11 @@ def generate(
25
 
26
 
27
  additional_inputs=[
28
- gr.Slider(
29
- label="Repetition penalty",
30
- value=1.2,
31
- minimum=1.0,
32
- maximum=2.0,
33
- step=0.05,
34
- interactive=True,
35
- info="Penalize repeated tokens",
36
- )
37
  ]
38
 
39
  examples=[["Expand the following prompt to add more detail: A storefront with 'Text to Image' written on it.", None, None ]]
 
25
 
26
 
27
  additional_inputs=[
28
+ gr.Slider(value=512, minimum=250, maximum=512, step=1, interactive=True, label="Max New Tokens"),
29
+ gr.Slider(value=1.2, minimum=0, maximum=2, step=0.05, interactive=True, label="Repetition Penalty"),
30
+ gr.Slider(value=0.5, minimum=0, maximum=1, step=0.05, interactive=True, label="Temperature"),
31
+ gr.Slider(value=1, minimum=0, maximum=2, step=0.05, interactive=True, label="Top P"),
32
+ gr.Slider(value=1, minimum=1, maximum=100, step=1, interactive=True, label="Top K"),
 
 
 
 
33
  ]
34
 
35
  examples=[["Expand the following prompt to add more detail: A storefront with 'Text to Image' written on it.", None, None ]]