Przeglądaj źródła

Further reorganize chat UI

oobabooga 2 lat temu
rodzic
commit
67d62475dc
1 zmienionych plików z 11 dodań i 10 usunięć
  1. 11 10
      server.py

+ 11 - 10
server.py

@@ -231,16 +231,7 @@ if shared.args.chat or shared.args.cai_chat:
 
 
             create_model_and_preset_menus()
             create_model_and_preset_menus()
 
 
-            with gr.Box():
-                with gr.Row():
-                    with gr.Column():
-                        shared.gradio['max_new_tokens'] = gr.Slider(minimum=shared.settings['max_new_tokens_min'], maximum=shared.settings['max_new_tokens_max'], step=1, label='max_new_tokens', value=shared.settings['max_new_tokens'])
-                        shared.gradio['chat_prompt_size_slider'] = gr.Slider(minimum=shared.settings['chat_prompt_size_min'], maximum=shared.settings['chat_prompt_size_max'], step=1, label='Maximum prompt size in tokens', value=shared.settings['chat_prompt_size'])
-                    with gr.Column():
-                        shared.gradio['chat_generation_attempts'] = gr.Slider(minimum=shared.settings['chat_generation_attempts_min'], maximum=shared.settings['chat_generation_attempts_max'], value=shared.settings['chat_generation_attempts'], step=1, label='Generation attempts (for longer replies)')
-                        shared.gradio['check'] = gr.Checkbox(value=shared.settings[f'stop_at_newline{suffix}'], label='Stop generating at new line character?')
-
-        with gr.Tab("Chat settings", elem_id="chat-settings"):
+        with gr.Tab("Character", elem_id="chat-settings"):
             shared.gradio['name1'] = gr.Textbox(value=shared.settings[f'name1{suffix}'], lines=1, label='Your name')
             shared.gradio['name1'] = gr.Textbox(value=shared.settings[f'name1{suffix}'], lines=1, label='Your name')
             shared.gradio['name2'] = gr.Textbox(value=shared.settings[f'name2{suffix}'], lines=1, label='Bot\'s name')
             shared.gradio['name2'] = gr.Textbox(value=shared.settings[f'name2{suffix}'], lines=1, label='Bot\'s name')
             shared.gradio['context'] = gr.Textbox(value=shared.settings[f'context{suffix}'], lines=5, label='Context')
             shared.gradio['context'] = gr.Textbox(value=shared.settings[f'context{suffix}'], lines=5, label='Context')
@@ -273,6 +264,16 @@ if shared.args.chat or shared.args.cai_chat:
                     shared.gradio['upload_img_tavern'] = gr.File(type='binary', file_types=['image'])
                     shared.gradio['upload_img_tavern'] = gr.File(type='binary', file_types=['image'])
 
 
         with gr.Tab("Settings", elem_id="settings"):
         with gr.Tab("Settings", elem_id="settings"):
+            with gr.Box():
+                gr.Markdown("Chat parameters")
+                with gr.Row():
+                    with gr.Column():
+                        shared.gradio['max_new_tokens'] = gr.Slider(minimum=shared.settings['max_new_tokens_min'], maximum=shared.settings['max_new_tokens_max'], step=1, label='max_new_tokens', value=shared.settings['max_new_tokens'])
+                        shared.gradio['chat_prompt_size_slider'] = gr.Slider(minimum=shared.settings['chat_prompt_size_min'], maximum=shared.settings['chat_prompt_size_max'], step=1, label='Maximum prompt size in tokens', value=shared.settings['chat_prompt_size'])
+                    with gr.Column():
+                        shared.gradio['chat_generation_attempts'] = gr.Slider(minimum=shared.settings['chat_generation_attempts_min'], maximum=shared.settings['chat_generation_attempts_max'], value=shared.settings['chat_generation_attempts'], step=1, label='Generation attempts (for longer replies)')
+                        shared.gradio['check'] = gr.Checkbox(value=shared.settings[f'stop_at_newline{suffix}'], label='Stop generating at new line character?')
+
             create_settings_menus(default_preset)
             create_settings_menus(default_preset)
 
 
         if shared.args.extensions is not None:
         if shared.args.extensions is not None: