423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-16 20:00:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: ui improve

Browse Source
This commit is contained in:
ZeYi Lin
2024-12-20 11:03:02 +08:00
parent 0a52962db3
commit 5f6dafd70e
5 changed files with 36 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/llamafactory/webui/components/train.py
View File

@@ -275,12 +275,11 @@ def create_train_tab(engine: "Engine") -> Dict[str, "Component"]:
use_swanlab = gr.Checkbox()
swanlab_project = gr.Textbox(value="LLaMA-Factory", placeholder="Project name", interactive=True)
swanlab_experiment_name = gr.Textbox(value="", placeholder="Experiment name", interactive=True)
swanlab_description = gr.Textbox(value="", placeholder="Experiment description", interactive=True)
swanlab_mode = gr.Dropdown(choices=["cloud", "local", "disabled"], value="cloud", interactive=True)
swanlab_workspace = gr.Textbox(value="", placeholder="Workspace name", interactive=True)
swanlab_api_key = gr.Textbox(value="", placeholder="API key", interactive=True)
swanlab_mode = gr.Dropdown(choices=["cloud", "local", "disabled"], value="cloud", interactive=True)
input_elems.update({use_swanlab, swanlab_api_key, swanlab_project, swanlab_workspace, swanlab_experiment_name, swanlab_description, swanlab_mode})
input_elems.update({use_swanlab, swanlab_api_key, swanlab_project, swanlab_workspace, swanlab_experiment_name, swanlab_mode})
elem_dict.update(
dict(
swanlab_tab=swanlab_tab,
@@ -289,7 +288,6 @@ def create_train_tab(engine: "Engine") -> Dict[str, "Component"]:
swanlab_project=swanlab_project,
swanlab_workspace=swanlab_workspace,
swanlab_experiment_name=swanlab_experiment_name,
swanlab_description=swanlab_description,
swanlab_mode=swanlab_mode,
)
)