Update g4f/models.py g4f/Provider/needs_auth/HuggingChat.py

kqlio67 · Dec 3, 2024 · b346510 · b346510
1 parent 8193daa
commit b346510
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/g4f/Provider/needs_auth/HuggingChat.py b/g4f/Provider/needs_auth/HuggingChat.py
@@ -25,6 +25,7 @@ class HuggingChat(AbstractProvider, ProviderModelMixin):
         'Qwen/Qwen2.5-72B-Instruct',
         'meta-llama/Meta-Llama-3.1-70B-Instruct',
         'CohereForAI/c4ai-command-r-plus-08-2024',
+        'Qwen/QwQ-32B-Preview',
         'nvidia/Llama-3.1-Nemotron-70B-Instruct-HF',
         'Qwen/Qwen2.5-Coder-32B-Instruct',
         'meta-llama/Llama-3.2-11B-Vision-Instruct',
@@ -37,6 +38,7 @@ class HuggingChat(AbstractProvider, ProviderModelMixin):
         "qwen-2.5-72b": "Qwen/Qwen2.5-72B-Instruct",
         "llama-3.1-70b": "meta-llama/Meta-Llama-3.1-70B-Instruct",
         "command-r-plus": "CohereForAI/c4ai-command-r-plus-08-2024",
+        "qwq-32b": "Qwen/QwQ-32B-Preview",
         "nemotron-70b": "nvidia/Llama-3.1-Nemotron-70B-Instruct-HF",
         "qwen-2.5-coder-32b": "Qwen/Qwen2.5-Coder-32B-Instruct",
         "llama-3.2-11b": "meta-llama/Llama-3.2-11B-Vision-Instruct",

diff --git a/g4f/models.py b/g4f/models.py
@@ -381,6 +381,12 @@ def __all__() -> list[str]:
     best_provider = IterListProvider([DeepInfraChat, HuggingChat, HuggingFace])
 )
 
+qwq_32b = Model(
+    name = 'qwq-32b',
+    base_provider = 'Qwen',
+    best_provider = IterListProvider([HuggingChat, HuggingFace])
+)
+
 ### Upstage ###
 solar_mini = Model(
     name = 'solar-mini',
@@ -762,6 +768,7 @@ class ModelUtils:
         # qwen 2.5
         'qwen-2.5-72b': qwen_2_5_72b,
         'qwen-2.5-coder-32b': qwen_2_5_coder_32b,
+        'qwq-32b': qwq_32b,
 
         ### Upstage ###
         'solar-mini': solar_mini,