pszemraj commited on
Commit
3688939
1 Parent(s): 7e755b6

⚡️ tweak params

Browse files

Signed-off-by: peter szemraj <peterszemraj@gmail.com>

Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -30,7 +30,7 @@ def proc_submission(
30
  length_penalty,
31
  repetition_penalty,
32
  no_repeat_ngram_size,
33
- max_input_length: int = 768,
34
  ):
35
  """
36
  proc_submission - a helper function for the gradio module to process submissions
@@ -209,7 +209,7 @@ if __name__ == "__main__":
209
  )
210
  with gr.Row():
211
  model_size = gr.Radio(
212
- choices=["base", "large"], label="Model Variant", value="large"
213
  )
214
  num_beams = gr.Radio(
215
  choices=[2, 3, 4],
@@ -228,9 +228,9 @@ if __name__ == "__main__":
228
  step=0.05,
229
  )
230
  token_batch_length = gr.Radio(
231
- choices=[512, 768, 1024],
232
  label="token batch length",
233
- value=512,
234
  )
235
 
236
  with gr.Row():
 
30
  length_penalty,
31
  repetition_penalty,
32
  no_repeat_ngram_size,
33
+ max_input_length: int = 1024,
34
  ):
35
  """
36
  proc_submission - a helper function for the gradio module to process submissions
 
209
  )
210
  with gr.Row():
211
  model_size = gr.Radio(
212
+ choices=["base", "large"], label="Model Variant", value="base"
213
  )
214
  num_beams = gr.Radio(
215
  choices=[2, 3, 4],
 
228
  step=0.05,
229
  )
230
  token_batch_length = gr.Radio(
231
+ choices=[512, 768, 1024, 1536],
232
  label="token batch length",
233
+ value=1024,
234
  )
235
 
236
  with gr.Row():