update webui

Former-commit-id: da30d0fb4abdb825f3383ddd106bb06a84695b7a

src/llmtuner/webui/components/top.py

@@ -4,7 +4,7 @@ import gradio as gr
 
 from llmtuner.extras.constants import METHODS, SUPPORTED_MODELS
 from llmtuner.extras.template import templates
-from llmtuner.webui.common import list_checkpoint, get_model_path, save_config, get_template
+from llmtuner.webui.common import list_checkpoint, get_model_path, get_template, save_config
 from llmtuner.webui.utils import can_quantize
 
 if TYPE_CHECKING:
@@ -36,10 +36,11 @@ def create_top() -> Dict[str, "Component"]:
         list_checkpoint, [model_name, finetuning_type], [checkpoints]
     ).then(
         get_model_path, [model_name], [model_path]
+    ).then(
+        get_template, [model_name], [template]
     ) # do not save config since the below line will save
 
     model_path.change(save_config, [lang, model_name, model_path])
-    model_path.change(get_template, [model_name], [template])
 
     finetuning_type.change(
         list_checkpoint, [model_name, finetuning_type], [checkpoints]

src/llmtuner/webui/components/train.py

@@ -3,10 +3,10 @@ from transformers.trainer_utils import SchedulerType
 
 import gradio as gr
 
+from llmtuner.extras.constants import STAGES
 from llmtuner.webui.common import list_checkpoint, list_dataset, DEFAULT_DATA_DIR
 from llmtuner.webui.components.data import create_preview_box
 from llmtuner.webui.utils import can_preview, get_preview, gen_plot
-from llmtuner.extras.constants import STAGES
 
 if TYPE_CHECKING:
     from gradio.components import Component
@@ -15,9 +15,7 @@ if TYPE_CHECKING:
 
 def create_train_tab(top_elems: Dict[str, "Component"], runner: "Runner") -> Dict[str, "Component"]:
     with gr.Row():
-        stage = gr.Dropdown(choices=STAGES,
-                            value="Supervised Finetuning", scale=2)
-
+        training_stage = gr.Dropdown(choices=STAGES, value=STAGES[0], scale=2)
         dataset_dir = gr.Textbox(value=DEFAULT_DATA_DIR, scale=2)
         dataset = gr.Dropdown(multiselect=True, scale=4)
         data_preview_btn = gr.Button(interactive=False, scale=1)
@@ -104,7 +102,7 @@ def create_train_tab(top_elems: Dict[str, "Component"], runner: "Runner") -> Dic
         top_elems["quantization_bit"],
         top_elems["template"],
         top_elems["source_prefix"],
-        stage,
+        training_stage,
         dataset_dir,
         dataset,
         max_source_length,
@@ -145,7 +143,7 @@ def create_train_tab(top_elems: Dict[str, "Component"], runner: "Runner") -> Dic
     )
 
     return dict(
-        stage=stage,
+        training_stage=training_stage,
         dataset_dir=dataset_dir,
         dataset=dataset,
         data_preview_btn=data_preview_btn,