423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-16 03:40:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

update ui

Browse Source 
Former-commit-id: 7f770f6895
This commit is contained in:
hiyouga
2024-07-03 23:13:49 +08:00
parent 28c8e083f4
commit b254df2d34
3 changed files with 23 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/llamafactory/webui/components/train.py
View File

@@ -95,12 +95,11 @@ def create_train_tab(engine: "Engine") -> Dict[str, "Component"]:
with gr.Row():
with gr.Column():
resize_vocab = gr.Checkbox()
packing = gr.Checkbox()
efficient_packing = gr.Checkbox()
neat_packing = gr.Checkbox()
with gr.Column():
upcast_layernorm = gr.Checkbox()
resize_vocab = gr.Checkbox()
use_llama_pro = gr.Checkbox()
with gr.Column():
@@ -114,10 +113,9 @@ def create_train_tab(engine: "Engine") -> Dict[str, "Component"]:
warmup_steps,
neftune_alpha,
optim,
resize_vocab,
packing,
efficient_packing,
upcast_layernorm,
neat_packing,
resize_vocab,
use_llama_pro,
shift_attn,
report_to,
@@ -131,10 +129,9 @@ def create_train_tab(engine: "Engine") -> Dict[str, "Component"]:
warmup_steps=warmup_steps,
neftune_alpha=neftune_alpha,
optim=optim,
resize_vocab=resize_vocab,
packing=packing,
efficient_packing=efficient_packing,
upcast_layernorm=upcast_layernorm,
neat_packing=neat_packing,
resize_vocab=resize_vocab,
use_llama_pro=use_llama_pro,
shift_attn=shift_attn,
report_to=report_to,
@@ -331,7 +328,7 @@ def create_train_tab(engine: "Engine") -> Dict[str, "Component"]:
)
dataset.focus(list_datasets, [dataset_dir, training_stage], [dataset], queue=False)
training_stage.change(change_stage, [training_stage], [dataset, packing, efficient_packing], queue=False)
training_stage.change(change_stage, [training_stage], [dataset, packing], queue=False)
reward_model.focus(list_checkpoints, [model_name, finetuning_type], [reward_model], queue=False)
model_name.change(list_output_dirs, [model_name, finetuning_type, current_time], [output_dir], queue=False)
finetuning_type.change(list_output_dirs, [model_name, finetuning_type, current_time], [output_dir], queue=False)