fix: bugs

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

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,