Nymbo commited on
Commit
ecec0b6
·
verified ·
1 Parent(s): 88d7bd2

Update ui_components.py

Browse files
Files changed (1) hide show
  1. 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 = 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,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
- "seed": (None if seed in (None, "", -1) else int(seed)),
 
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]