lucas
/
LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use default arg for freeze tuning

This commit is contained in:
hiyouga 2024-03-09 06:08:48 +08:00
parent 818726e9bc
commit af0e370fb1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llmtuner/webui/components/train.py
View File

@ -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.Accordion(label="Freeze config", open=False) as freeze_tab:
with gr.Row(): with gr.Row():
num_layer_trainable = gr.Slider(value=3, minimum=1, maximum=128, step=1, scale=2) 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}) input_elems.update({num_layer_trainable, name_module_trainable})
elem_dict.update( elem_dict.update(