ros/LlamaFactory
1
0
Fork 0
mirror of https://github.com/hiyouga/LlamaFactory.git synced 2026-02-01 08:13:38 +00:00
Code Issues 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

5
src/llmtuner/webui/common.py
View File

@@ -9,6 +9,7 @@ from ..extras.constants import (
DATA_CONFIG,
DEFAULT_MODULE,
DEFAULT_TEMPLATE,
MLLM_LIST,
PEFT_METHODS,
STAGES_USE_PAIR_DATA,
SUPPORTED_MODELS,
@@ -105,6 +106,10 @@ def get_template(model_name: str) -> str:
return "default"
def get_visual(model_name: str) -> bool:
return get_prefix(model_name) in MLLM_LIST
def list_adapters(model_name: str, finetuning_type: str) -> "gr.Dropdown":
if finetuning_type not in PEFT_METHODS:
return gr.Dropdown(value=[], choices=[], interactive=False)