Support Qwen1.5 [release]

datadreamer-dev · Apr 4, 2024 · a3a22b6 · a3a22b6
1 parent 792ce1c
commit a3a22b6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "DataDreamer"
-version = "0.26.0"
+version = "0.27.0"
 description = "Prompt. Generate Synthetic Data. Train & Align Models."
 license = "MIT"
 authors= [

diff --git a/src/llms/_chat_prompt_templates.py b/src/llms/_chat_prompt_templates.py
@@ -166,6 +166,8 @@ def _model_name_to_chat_prompt_template_type(  # noqa: C901
             chat_prompt_template_type = "chatml_system"
         elif all(fragment in model_name_lower for fragment in ["qwen-", "-chat"]):
             chat_prompt_template_type = "chatml_system"
+        elif all(fragment in model_name_lower for fragment in ["qwen1.5-", "-chat"]):
+            chat_prompt_template_type = "chatml_system"
         elif all(fragment in model_name_lower for fragment in ["dolphin-", "-mistral"]):
             chat_prompt_template_type = "chatml_system"
         elif all(