Update ui_components.py
Browse files- ui_components.py +3 -2
ui_components.py
CHANGED
|
@@ -141,7 +141,7 @@ class UIComponents:
|
|
| 141 |
self.top_p_slider = gr.Slider(minimum=0.0, maximum=1.0, value=1.0, step=0.01, label="Top-p")
|
| 142 |
with gr.Row():
|
| 143 |
self.max_tokens_box = gr.Number(value=8192, precision=0, label="Max tokens")
|
| 144 |
-
self.seed_box = gr.Number(value=None, precision=0, label="Seed (-1
|
| 145 |
with gr.Row():
|
| 146 |
self.frequency_penalty = gr.Slider(minimum=-2.0, maximum=2.0, value=0.0, step=0.01, label="Frequency penalty")
|
| 147 |
self.presence_penalty = gr.Slider(minimum=-2.0, maximum=2.0, value=0.0, step=0.01, label="Presence penalty")
|
|
@@ -181,7 +181,8 @@ class UIComponents:
|
|
| 181 |
"temperature": float(temperature) if temperature is not None else None,
|
| 182 |
"top_p": float(top_p) if top_p is not None else None,
|
| 183 |
"max_tokens": int(max_tokens) if max_tokens else None,
|
| 184 |
-
|
|
|
|
| 185 |
"frequency_penalty": float(frequency_penalty) if frequency_penalty is not None else None,
|
| 186 |
"presence_penalty": float(presence_penalty) if presence_penalty is not None else None,
|
| 187 |
# stop: list[str]
|
|
|
|
| 141 |
self.top_p_slider = gr.Slider(minimum=0.0, maximum=1.0, value=1.0, step=0.01, label="Top-p")
|
| 142 |
with gr.Row():
|
| 143 |
self.max_tokens_box = gr.Number(value=8192, precision=0, label="Max tokens")
|
| 144 |
+
self.seed_box = gr.Number(value=None, precision=0, label="Seed (-1 for random)")
|
| 145 |
with gr.Row():
|
| 146 |
self.frequency_penalty = gr.Slider(minimum=-2.0, maximum=2.0, value=0.0, step=0.01, label="Frequency penalty")
|
| 147 |
self.presence_penalty = gr.Slider(minimum=-2.0, maximum=2.0, value=0.0, step=0.01, label="Presence penalty")
|
|
|
|
| 181 |
"temperature": float(temperature) if temperature is not None else None,
|
| 182 |
"top_p": float(top_p) if top_p is not None else None,
|
| 183 |
"max_tokens": int(max_tokens) if max_tokens else None,
|
| 184 |
+
# seed: -1 means random (omit from payload)
|
| 185 |
+
"seed": (None if (seed in (-1, "-1")) else (int(seed) if seed not in (None, "") else None)),
|
| 186 |
"frequency_penalty": float(frequency_penalty) if frequency_penalty is not None else None,
|
| 187 |
"presence_penalty": float(presence_penalty) if presence_penalty is not None else None,
|
| 188 |
# stop: list[str]
|