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

improve llamaboard

This commit is contained in:
hiyouga 2024-06-19 23:46:03 +08:00
parent 3f84411b5d
commit f22d8f9ca4
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/llamafactory/webui/components/top.py
View File

@ -50,9 +50,13 @@ def create_top() -> Dict[str, "Component"]:
visual_inputs = gr.Checkbox(scale=1) visual_inputs = gr.Checkbox(scale=1)
model_name.change(get_model_info, [model_name], [model_path, template, visual_inputs], queue=False) model_name.change(get_model_info, [model_name], [model_path, template, visual_inputs], queue=False)
model_name.input(save_config, inputs=[lang, model_name], queue=False) model_name.input(save_config, inputs=[lang, model_name], queue=False).then(
list_checkpoints, [model_name, finetuning_type], [checkpoint_path], queue=False
)
model_path.input(save_config, inputs=[lang, model_name, model_path], queue=False) model_path.input(save_config, inputs=[lang, model_name, model_path], queue=False)
finetuning_type.change(can_quantize, [finetuning_type], [quantization_bit], queue=False) finetuning_type.change(can_quantize, [finetuning_type], [quantization_bit], queue=False).then(
list_checkpoints, [model_name, finetuning_type], [checkpoint_path], queue=False
)
checkpoint_path.focus(list_checkpoints, [model_name, finetuning_type], [checkpoint_path], queue=False) checkpoint_path.focus(list_checkpoints, [model_name, finetuning_type], [checkpoint_path], queue=False)
return dict( return dict(