💄 style(ollama): Phi3 Instruct models and its model icons (#2254)

lobehub · Apr 28, 2024 · c9b55cc · c9b55cc
1 parent e5cc179
commit c9b55cc
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/src/components/ModelIcon/index.tsx b/src/components/ModelIcon/index.tsx
@@ -95,6 +95,7 @@ const ModelIcon = memo<ModelProviderIconProps>(({ model: originModel, size = 12
     return <Stability.Avatar size={size} />;
 
   if (model.includes('wizardlm')) return <Azure.Avatar size={size} />;
+  if (model.includes('phi3')) return <Azure.Avatar size={size} />;
   if (model.includes('firefly')) return <Adobe.Avatar size={size} />;
   if (model.includes('jamba') ||
     model.includes('j2-'))

diff --git a/src/components/ModelTag/ModelIcon.tsx b/src/components/ModelTag/ModelIcon.tsx
@@ -82,6 +82,7 @@ const ModelIcon = memo<ModelIconProps>(({ model, size = 12 }) => {
     return <Stability size={size} />;
 
   if (model.includes('wizardlm')) return <Azure size={size} />;
+  if (model.includes('phi3')) return <Azure size={size} />;
   if (model.includes('firefly')) return <AdobeFirefly size={size} />;
   if (model.includes('jamba') || model.includes('j2-')) return <Ai21 size={size} />;
 });

diff --git a/src/config/modelProviders/ollama.ts b/src/config/modelProviders/ollama.ts
@@ -91,6 +91,12 @@ const Ollama: ModelProviderCard = {
       id: 'codellama:python',
       tokens: 16_000,
     },
+    {
+      displayName: 'Phi3-Instruct 3.8B',
+      enabled: true,
+      id: 'phi3:instruct',
+      tokens: 128_000,
+    },
     {
       displayName: 'Mistral',
       enabled: true,