Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -85,7 +85,7 @@ def generate(
|
|
| 85 |
{"input_ids": input_ids},
|
| 86 |
streamer=streamer,
|
| 87 |
max_new_tokens=max_new_tokens,
|
| 88 |
-
do_sample=True,
|
| 89 |
top_p=top_p,
|
| 90 |
top_k=top_k,
|
| 91 |
temperature=temperature,
|
|
@@ -113,31 +113,31 @@ chat_interface = gr.ChatInterface(
|
|
| 113 |
),
|
| 114 |
gr.Slider(
|
| 115 |
label="Temperature",
|
| 116 |
-
minimum=0.
|
| 117 |
maximum=4.0,
|
| 118 |
step=0.1,
|
| 119 |
-
value=0.
|
| 120 |
),
|
| 121 |
gr.Slider(
|
| 122 |
label="Top-p (nucleus sampling)",
|
| 123 |
minimum=0.05,
|
| 124 |
maximum=1.0,
|
| 125 |
step=0.05,
|
| 126 |
-
value=0.
|
| 127 |
),
|
| 128 |
gr.Slider(
|
| 129 |
label="Top-k",
|
| 130 |
minimum=1,
|
| 131 |
maximum=1000,
|
| 132 |
step=1,
|
| 133 |
-
value=
|
| 134 |
),
|
| 135 |
gr.Slider(
|
| 136 |
label="Repetition penalty",
|
| 137 |
minimum=1.0,
|
| 138 |
maximum=2.0,
|
| 139 |
step=0.05,
|
| 140 |
-
value=1.
|
| 141 |
),
|
| 142 |
],
|
| 143 |
stop_btn=None,
|
|
|
|
| 85 |
{"input_ids": input_ids},
|
| 86 |
streamer=streamer,
|
| 87 |
max_new_tokens=max_new_tokens,
|
| 88 |
+
do_sample=False if temperature == 0.0 else True,
|
| 89 |
top_p=top_p,
|
| 90 |
top_k=top_k,
|
| 91 |
temperature=temperature,
|
|
|
|
| 113 |
),
|
| 114 |
gr.Slider(
|
| 115 |
label="Temperature",
|
| 116 |
+
minimum=0.0,
|
| 117 |
maximum=4.0,
|
| 118 |
step=0.1,
|
| 119 |
+
value=0.0,
|
| 120 |
),
|
| 121 |
gr.Slider(
|
| 122 |
label="Top-p (nucleus sampling)",
|
| 123 |
minimum=0.05,
|
| 124 |
maximum=1.0,
|
| 125 |
step=0.05,
|
| 126 |
+
value=0.95,
|
| 127 |
),
|
| 128 |
gr.Slider(
|
| 129 |
label="Top-k",
|
| 130 |
minimum=1,
|
| 131 |
maximum=1000,
|
| 132 |
step=1,
|
| 133 |
+
value=20,
|
| 134 |
),
|
| 135 |
gr.Slider(
|
| 136 |
label="Repetition penalty",
|
| 137 |
minimum=1.0,
|
| 138 |
maximum=2.0,
|
| 139 |
step=0.05,
|
| 140 |
+
value=1.0,
|
| 141 |
),
|
| 142 |
],
|
| 143 |
stop_btn=None,
|