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

add test mm plugin

Browse Source 
Former-commit-id: a2a8c0b92c
This commit is contained in:
hiyouga
2024-08-31 01:53:38 +08:00
parent f31e7e0dfc
commit 09a2ecebc4
5 changed files with 192 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/llamafactory/webui/components/export.py
View File

@@ -46,7 +46,6 @@ def save_model(
finetuning_type: str,
checkpoint_path: Union[str, List[str]],
template: str,
visual_inputs: bool,
export_size: int,
export_quantization_bit: str,
export_quantization_dataset: str,
@@ -78,7 +77,6 @@ def save_model(
model_name_or_path=model_path,
finetuning_type=finetuning_type,
template=template,
visual_inputs=visual_inputs,
export_dir=export_dir,
export_hub_model_id=export_hub_model_id or None,
export_size=export_size,
@@ -129,7 +127,6 @@ def create_export_tab(engine: "Engine") -> Dict[str, "Component"]:
engine.manager.get_elem_by_id("top.finetuning_type"),
engine.manager.get_elem_by_id("top.checkpoint_path"),
engine.manager.get_elem_by_id("top.template"),
engine.manager.get_elem_by_id("top.visual_inputs"),
export_size,
export_quantization_bit,
export_quantization_dataset,