423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-24 06:42:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: bugs

Browse Source 
Former-commit-id: d0eb64d5e3472a166c9adac4cb4ba06bdd663e46
This commit is contained in:
ZeYi Lin 2024-12-19 21:08:16 +08:00
parent b512a06c3d
commit dd22454fc5
3 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/llamafactory/hparams/finetuning_args.py
View File

@ -334,10 +334,6 @@ class SwanLabArguments:
default=None, default=None,
metadata={"help": "The API key for SwanLab."}, metadata={"help": "The API key for SwanLab."},
) )
swanlab_config: dict = field(
default={"Framework": "🦙LLaMA Factory"},
metadata={"help": "The configuration file for SwanLab."},
)
@dataclass @dataclass

1
src/llamafactory/train/trainer_utils.py
View File

@ -475,6 +475,7 @@ def get_swanlab_callback(finetuning_args: "FinetuningArguments") -> "TrainerCall
experiment_name=finetuning_args.swanlab_experiment_name, experiment_name=finetuning_args.swanlab_experiment_name,
description=finetuning_args.swanlab_description, description=finetuning_args.swanlab_description,
mode=finetuning_args.swanlab_mode, mode=finetuning_args.swanlab_mode,
config={"Framework": "🦙LLaMA Factory"},
) )
return swanlab_callback return swanlab_callback

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

@ -280,7 +280,7 @@ def create_train_tab(engine: "Engine") -> Dict[str, "Component"]:
swanlab_description = gr.Textbox(value="", placeholder="Experiment description", 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_mode = gr.Dropdown(choices=["cloud", "local", "disabled"], value="cloud", interactive=True)
input_elems.update({use_swanlab}) input_elems.update({use_swanlab, swanlab_api_key, swanlab_project, swanlab_workspace, swanlab_experiment_name, swanlab_description, swanlab_mode})
elem_dict.update( elem_dict.update(
dict( dict(
swanlab_tab=swanlab_tab, swanlab_tab=swanlab_tab,