Update hf_adapter.py

Signed-off-by: gitworkflows <[email protected]>
khulnasoft · Oct 12, 2024 · 2a80971 · 2a80971
1 parent fe09598
commit 2a80971
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/gptdb/model/adapter/hf_adapter.py b/gptdb/model/adapter/hf_adapter.py
@@ -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
+            )
         )