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

add llava to llamaboard

Browse Source 
Former-commit-id: deaaff0a9de0eef9691991c99cd797461b1165cc
This commit is contained in:
hiyouga
2024-04-26 06:41:35 +08:00
parent 1480e3a88f
commit 697bc20941
13 changed files with 66 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -27,6 +27,7 @@ def save_model(
adapter_path: List[str],
finetuning_type: str,
template: str,
visual_inputs: bool,
export_size: int,
export_quantization_bit: int,
export_quantization_dataset: str,
@@ -66,6 +67,7 @@ def save_model(
adapter_name_or_path=adapter_name_or_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,
@@ -105,6 +107,7 @@ def create_export_tab(engine: "Engine") -> Dict[str, "Component"]:
engine.manager.get_elem_by_id("top.adapter_path"),
engine.manager.get_elem_by_id("top.finetuning_type"),
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,