mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-24 06:42:52 +08:00
fix: bugs
Former-commit-id: d0eb64d5e3472a166c9adac4cb4ba06bdd663e46
This commit is contained in:
parent
b512a06c3d
commit
dd22454fc5
@ -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
|
||||
|
@ -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
|
@ -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,
|
||||
|
Loading…
x
Reference in New Issue
Block a user