fix: bugs

Former-commit-id: a2297f97f7587c77d55fbce9ffa81dc60d0b04a1
This commit is contained in:
ZeYi Lin 2024-12-19 21:08:16 +08:00
parent a30bdbb1c0
commit 925e421bde
3 changed files with 2 additions and 5 deletions

View File

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

View File

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

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_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(
dict(
swanlab_tab=swanlab_tab,