add Yi-VL-34B model

2024-05-15 22:58:19 +08:00
parent 73845fcc46
commit a388cadfc0
5 changed files with 15 additions and 5 deletions
--- a/src/llmtuner/extras/constants.py
+++ b/src/llmtuner/extras/constants.py
@@ -1215,6 +1215,9 @@ register_model_group(
        "YiVL-6B-Chat": {
            DownloadSource.DEFAULT: "BUAADreamer/Yi-VL-6B-hf",
        },
+        "YiVL-34B-Chat": {
+            DownloadSource.DEFAULT: "BUAADreamer/Yi-VL-34B-hf",
+        },
    },
    template="yi_vl",
    vision=True,