github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-05 02:51:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(model): Support Qwen2.5 models (#2030)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-09-19 11:58:39 +08:00
committed by GitHub
parent 1328c5636d
commit 1f2e844639
5 changed files with 47 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
dbgpt/model/adapter/hf_adapter.py
View File

@@ -370,10 +370,15 @@ class Qwen2Adapter(QwenAdapter):
support_8bit: bool = True
def do_match(self, lower_model_name_or_path: Optional[str] = None):
return (
lower_model_name_or_path
and "qwen2" in lower_model_name_or_path
and "instruct" in lower_model_name_or_path
return lower_model_name_or_path and (
(
"qwen2" in lower_model_name_or_path
and "instruct" in lower_model_name_or_path
)
or (
"qwen2.5" in lower_model_name_or_path
and "instruct" in lower_model_name_or_path
)
)