Leyo commited on
Commit
c586e09
1 Parent(s): e3795af

2048 max len + label fix

Browse files
Files changed (1) hide show
  1. app_dialogue.py +10 -10
app_dialogue.py CHANGED
@@ -318,7 +318,7 @@ with gr.Blocks(title="IDEFICS", theme=gr.themes.Base()) as demo:
318
  )
319
  max_new_tokens = gr.Slider(
320
  minimum=0,
321
- maximum=1024,
322
  value=512,
323
  step=64,
324
  interactive=True,
@@ -338,7 +338,7 @@ with gr.Blocks(title="IDEFICS", theme=gr.themes.Base()) as demo:
338
  value=0.0,
339
  step=1.0,
340
  interactive=True,
341
- label="No repeat ngram size",
342
  )
343
  length_penalty = gr.Slider(
344
  minimum=0.0,
@@ -348,14 +348,6 @@ with gr.Blocks(title="IDEFICS", theme=gr.themes.Base()) as demo:
348
  interactive=True,
349
  label="Length penalty",
350
  )
351
- no_repeat_ngram_size = gr.Slider(
352
- minimum=0.0,
353
- maximum=10.0,
354
- value=0.0,
355
- step=1.0,
356
- interactive=True,
357
- label="No repeat ngram size",
358
- )
359
  penalty_alpha = gr.Slider(
360
  minimum=0.0,
361
  maximum=5.0,
@@ -364,6 +356,14 @@ with gr.Blocks(title="IDEFICS", theme=gr.themes.Base()) as demo:
364
  interactive=True,
365
  label="Penalty alpha",
366
  )
 
 
 
 
 
 
 
 
367
 
368
  with gr.Column(scale=6):
369
  chatbot = gr.Chatbot(
 
318
  )
319
  max_new_tokens = gr.Slider(
320
  minimum=0,
321
+ maximum=2048,
322
  value=512,
323
  step=64,
324
  interactive=True,
 
338
  value=0.0,
339
  step=1.0,
340
  interactive=True,
341
+ label="Minimum generation length",
342
  )
343
  length_penalty = gr.Slider(
344
  minimum=0.0,
 
348
  interactive=True,
349
  label="Length penalty",
350
  )
 
 
 
 
 
 
 
 
351
  penalty_alpha = gr.Slider(
352
  minimum=0.0,
353
  maximum=5.0,
 
356
  interactive=True,
357
  label="Penalty alpha",
358
  )
359
+ no_repeat_ngram_size = gr.Slider(
360
+ minimum=0.0,
361
+ maximum=10.0,
362
+ value=0.0,
363
+ step=1.0,
364
+ interactive=True,
365
+ label="No repeat ngram size",
366
+ )
367
 
368
  with gr.Column(scale=6):
369
  chatbot = gr.Chatbot(