💄 style: Add HuggingFace Model: DeepSeek R1 (#5564)

lobehub · Jan 23, 2025 · 66d4edd · 66d4edd
1 parent 223e3e4
commit 66d4edd
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/src/config/aiModels/huggingface.ts b/src/config/aiModels/huggingface.ts
@@ -5,7 +5,6 @@ const huggingfaceChatModels: AIChatModelCard[] = [
     contextWindowTokens: 32_768,
     description: 'Mistral AI的指令调优模型',
     displayName: 'Mistral 7B Instruct v0.3',
-    enabled: true,
     id: 'mistralai/Mistral-7B-Instruct-v0.3',
     type: 'chat',
   },
@@ -34,6 +33,7 @@ const huggingfaceChatModels: AIChatModelCard[] = [
     contextWindowTokens: 32_768,
     description: 'Qwen QwQ 是由 Qwen 团队开发的实验研究模型，专注于提升AI推理能力。',
     displayName: 'QwQ 32B Preview',
+    enabled: true,
     id: 'Qwen/QwQ-32B-Preview',
     type: 'chat',
   },
@@ -49,6 +49,13 @@ const huggingfaceChatModels: AIChatModelCard[] = [
     id: 'NousResearch/Hermes-3-Llama-3.1-8B',
     type: 'chat',
   },
+  {
+    contextWindowTokens: 16_384,
+    displayName: 'DeepSeek R1',
+    enabled: true,
+    id: 'deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',
+    type: 'chat',
+  },
 ];
 
 export const allModels = [...huggingfaceChatModels];

diff --git a/src/config/modelProviders/huggingface.ts b/src/config/modelProviders/huggingface.ts
@@ -6,7 +6,6 @@ const HuggingFace: ModelProviderCard = {
       contextWindowTokens: 32_768,
       description: 'Mistral AI的指令调优模型',
       displayName: 'Mistral 7B Instruct v0.3',
-      enabled: true,
       id: 'mistralai/Mistral-7B-Instruct-v0.3',
     },
     {
@@ -31,6 +30,7 @@ const HuggingFace: ModelProviderCard = {
       contextWindowTokens: 32_768,
       description: 'Qwen QwQ 是由 Qwen 团队开发的实验研究模型，专注于提升AI推理能力。',
       displayName: 'QwQ 32B Preview',
+      enabled: true,
       id: 'Qwen/QwQ-32B-Preview',
     },
     {
@@ -43,6 +43,12 @@ const HuggingFace: ModelProviderCard = {
       displayName: 'Hermes 3 Llama 3.1 8B',
       id: 'NousResearch/Hermes-3-Llama-3.1-8B',
     },
+    {
+      contextWindowTokens: 16_384,
+      displayName: 'DeepSeek R1',
+      enabled: true,
+      id: 'deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',
+    },
   ],
   checkModel: 'mistralai/Mistral-7B-Instruct-v0.2',
   description: