Former-commit-id: 81176fe226da89eace89cb202bad68e73b7c2a02
This commit is contained in:
fzc8578
2025-01-04 11:11:15 +08:00
parent 5504b5254c
commit 2c120aa0df
7 changed files with 164 additions and 2 deletions

View File

@@ -1141,6 +1141,17 @@ register_model_group(
)
register_model_group(
models={
"MiniCPM-V-2_6-Chat": {
DownloadSource.DEFAULT: "openbmb/MiniCPM-V-2_6",
DownloadSource.MODELSCOPE: "OpenBMB/MiniCPM-V-2_6",
},
},
template="cpm_o",
)
register_model_group(
models={
"Mistral-7B-v0.1": {

View File

@@ -81,7 +81,7 @@ def check_dependencies() -> None:
logger.warning_once("Version checking has been disabled, may lead to unexpected behaviors.")
return
require_version("transformers>=4.41.2,<=4.46.1", "To fix: pip install transformers>=4.41.2,<=4.46.1")
require_version("transformers>=4.41.2", "To fix: pip install transformers>=4.41.2")
require_version("datasets>=2.16.0,<=3.1.0", "To fix: pip install datasets>=2.16.0,<=3.1.0")
require_version("accelerate>=0.34.0,<=1.0.1", "To fix: pip install accelerate>=0.34.0,<=1.0.1")
require_version("peft>=0.11.1,<=0.12.0", "To fix: pip install peft>=0.11.1,<=0.12.0")