From 8f6eb1383d4cb50def071c9aefa5b77b9bf21258 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Sat, 9 Mar 2024 06:08:48 +0800 Subject: [PATCH] use default arg for freeze tuning Former-commit-id: af0e370fb16f3e0cf2f4c8036301d5253d8249b9 --- src/llmtuner/webui/components/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llmtuner/webui/components/train.py b/src/llmtuner/webui/components/train.py index 4bcbb967..25a518b9 100644 --- a/src/llmtuner/webui/components/train.py +++ b/src/llmtuner/webui/components/train.py @@ -116,7 +116,7 @@ def create_train_tab(engine: "Engine") -> Dict[str, "Component"]: with gr.Accordion(label="Freeze config", open=False) as freeze_tab: with gr.Row(): num_layer_trainable = gr.Slider(value=3, minimum=1, maximum=128, step=1, scale=2) - name_module_trainable = gr.Textbox(scale=3) + name_module_trainable = gr.Textbox(value="all", scale=3) input_elems.update({num_layer_trainable, name_module_trainable}) elem_dict.update(