Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -25,15 +25,11 @@ def generate(
|
|
25 |
|
26 |
|
27 |
additional_inputs=[
|
28 |
-
gr.Slider(
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
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 ]]
|