slush0 commited on
Commit
512c082
·
1 Parent(s): af5243f

Allow also max_lenght=64.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -76,7 +76,7 @@ with gr.Blocks() as iface:
76
  value=["\\n", "</s>"], label='Extra end sequences')
77
 
78
  # Maximum length of inference session
79
- max_length = gr.Radio([128, 256, 512, 1024, 2048], value=512, interactive=True, label="Max length")
80
 
81
  with gr.Row():
82
  with gr.Column():
@@ -84,7 +84,7 @@ with gr.Blocks() as iface:
84
  do_sample = gr.Checkbox(value=True, interactive=True, label="do_sample")
85
 
86
  # Should the app append stop sequence at the end of prompt or should it leave the prompt open?
87
- add_stoptoken = gr.Checkbox(value=True, interactive=True, label="Automatically add stop token to prompt.")
88
 
89
  # Only one of top_k and top_p can be set. Requires "do_sample=True" to work.
90
  top_k = gr.Number(value=0, precision=0, interactive=True, label="top_k")
 
76
  value=["\\n", "</s>"], label='Extra end sequences')
77
 
78
  # Maximum length of inference session
79
+ max_length = gr.Radio([64, 128, 256, 512, 1024, 2048], value=256, interactive=True, label="Max length")
80
 
81
  with gr.Row():
82
  with gr.Column():
 
84
  do_sample = gr.Checkbox(value=True, interactive=True, label="do_sample")
85
 
86
  # Should the app append stop sequence at the end of prompt or should it leave the prompt open?
87
+ add_stoptoken = gr.Checkbox(value=True, interactive=True, label="Automatically add eos token to the prompt.")
88
 
89
  # Only one of top_k and top_p can be set. Requires "do_sample=True" to work.
90
  top_k = gr.Number(value=0, precision=0, interactive=True, label="top_k")