add: temporally add llama-3.2-11B-Vision-Instruct on top of the list

lablup · Jan 7, 2025 · 336a5f5 · 336a5f5
1 parent 8a10f22
commit 336a5f5
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/react/src/components/ModelCardModal.tsx b/react/src/components/ModelCardModal.tsx
@@ -167,9 +167,11 @@ const ModelCardModal: React.FC<ModelCardModalProps> = ({
                           : 'v1'
                       }
                       modelName={
-                        model_card?.name?.includes('stable-diffusion-3-medium')
+                        model_card?.name === 'stable-diffusion-3-medium'
                           ? 'stable-diffusion-3m'
-                          : model_card?.name || ''
+                          : model_card?.name === 'Llama-3.2-11B-Vision-Instruct'
+                            ? 'llama-vision-11b'
+                            : model_card?.name || ''
                       }
                     />
                   </Flex>

diff --git a/react/src/components/ModelTryContent.tsx b/react/src/components/ModelTryContent.tsx
@@ -205,7 +205,9 @@ const ModelTryContent: React.FC<ModelTryContentProps> = ({
   ): ServiceLauncherFormValue => {
     const model = modelName?.includes('stable-diffusion-3-medium')
       ? 'stable-diffusion-3m'
-      : modelName;
+      : modelName?.includes('Llama-3.2-11B-Vision-Instruct')
+        ? 'llama-vision-11b'
+        : modelName;
     return {
       serviceName: `${model}-${generateRandomString(4)}`,
       desiredRoutingCount: 1,
@@ -537,7 +539,8 @@ const ModelTryContent: React.FC<ModelTryContentProps> = ({
       <Button
         disabled={
           modelName?.includes('stable-diffusion') ||
-          modelName?.includes('gemma-2-27b-it')
+          modelName?.includes('gemma-2-27b-it') ||
+          modelName?.includes('Llama-3.2-11B-Vision-Instruct')
         }
         type="primary"
         onClick={() => {

diff --git a/react/src/pages/ModelStoreListPage.tsx b/react/src/pages/ModelStoreListPage.tsx
@@ -244,6 +244,7 @@ const ModelStoreListPage: React.FC = () => {
           .sort((a, b) => {
             const specialNames = [
               'gemma-2-27b-it',
+              'Llama-3.2-11B-Vision-Instruct',
               'stable-diffusion-3-medium',
             ];
             const aIndex = specialNames.indexOf(a?.name || '');