Просмотр исходного кода

Better default for chat output length

Ideally, generation should stop at '\n', but this feature is brand new
on transformers (https://github.com/huggingface/transformers/pull/20727)
oobabooga 3 лет назад
Родитель
Сommit
b871f76aac
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      server.py

+ 1 - 1
server.py

@@ -181,7 +181,7 @@ elif args.chat:
             with gr.Column():
                 with gr.Row(equal_height=True):
                     with gr.Column():
-                        length_slider = gr.Slider(minimum=1, maximum=2000, step=1, label='max_length', value=200)
+                        length_slider = gr.Slider(minimum=1, maximum=2000, step=1, label='max_length', value=100)
                         preset_menu = gr.Dropdown(choices=available_presets, value="NovelAI-Sphinx Moth", label='Preset')
                     with gr.Column():
                         temp_slider = gr.Slider(minimum=0.0, maximum=1.0, step=0.01, label='Temperature', value=0.7)