add llava to llamaboard

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

View File

@@ -28,6 +28,8 @@ LOG_FILE_NAME = "trainer_log.jsonl"
METHODS = ["full", "freeze", "lora"]
MLLM_LIST = ["LLaVA1.5"]
MOD_SUPPORTED_MODELS = ["bloom", "falcon", "gemma", "llama", "mistral", "mixtral", "phi", "starcoder2"]
PEFT_METHODS = ["lora"]
@@ -566,6 +568,19 @@ register_model_group(
)
register_model_group(
models={
"LLaVA1.5-7B-Chat": {
DownloadSource.DEFAULT: "llava-hf/llava-1.5-7b-hf",
},
"LLaVA1.5-13B-Chat": {
DownloadSource.DEFAULT: "llava-hf/llava-1.5-13b-hf",
},
},
template="vicuna",
)
register_model_group(
models={
"Mistral-7B-v0.1": {