Spaces:
Runtime error
Runtime error
Allow also max_lenght=64.
Browse files
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=
|
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
|
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")
|