add swanlab

This commit is contained in:
hiyouga
2024-12-19 07:12:31 +00:00
parent 6ccd64ecd9
commit 96f8f103e5
6 changed files with 60 additions and 3 deletions

View File

@@ -270,6 +270,18 @@ def create_train_tab(engine: "Engine") -> Dict[str, "Component"]:
)
)
with gr.Accordion(open=False) as swanlab_tab:
with gr.Row():
use_swanlab = gr.Checkbox()
input_elems.update({use_swanlab})
elem_dict.update(
dict(
swanlab_tab=swanlab_tab,
use_swanlab=use_swanlab,
)
)
with gr.Row():
cmd_preview_btn = gr.Button()
arg_save_btn = gr.Button()