diff --git a/react/src/components/ModelTryContent.tsx b/react/src/components/ModelTryContent.tsx index d975324ae..ac3facc27 100644 --- a/react/src/components/ModelTryContent.tsx +++ b/react/src/components/ModelTryContent.tsx @@ -220,7 +220,7 @@ const ModelTryContent: React.FC = ({ image: { registry: 'cr.backend.ai', name: (() => { - if (modelName?.includes('stable-diffusion')) { + if (['stable-diffusion', 'phi-4'].includes(modelName as string)) { return 'testing/ngc-pytorch'; } switch (runtimeVariant) { @@ -233,7 +233,7 @@ const ModelTryContent: React.FC = ({ } })(), tag: (() => { - if (modelName?.includes('stable-diffusion')) { + if (['stable-diffusion', 'phi-4'].includes(modelName as string)) { return '24.07-pytorch2.4-py310-cuda12.5'; } switch (runtimeVariant) { @@ -264,7 +264,9 @@ const ModelTryContent: React.FC = ({ version: '', }, // FIXME: temporally hard-coded runtime variant - runtimeVariant: modelName?.includes('stable-diffusion') + runtimeVariant: ['stable-diffusion', 'phi-4'].includes( + modelName as string, + ) ? 'custom' : runtimeVariant, cluster_size: 1, @@ -536,7 +538,8 @@ const ModelTryContent: React.FC = ({ type="primary" disabled={ modelName?.includes('stable-diffusion') || - modelName?.includes('Talkativot UI') + modelName?.includes('Talkativot UI') || + modelName?.includes('phi-4') } onClick={() => { cloneOrCreateModelService('vllm'); @@ -553,7 +556,8 @@ const ModelTryContent: React.FC = ({ modelName?.includes('stable-diffusion') || modelName?.includes('gemma-2-27b-it') || modelName?.includes('Llama-3.2-11B-Vision-Instruct') || - modelName?.includes('Talkativot UI') + modelName?.includes('Talkativot UI') || + modelName?.includes('phi-4') } type="primary" onClick={() => { diff --git a/react/src/pages/ModelStoreListPage.tsx b/react/src/pages/ModelStoreListPage.tsx index d342b97ec..44139d1e6 100644 --- a/react/src/pages/ModelStoreListPage.tsx +++ b/react/src/pages/ModelStoreListPage.tsx @@ -243,6 +243,7 @@ const ModelStoreListPage: React.FC = () => { }) .sort((a, b) => { const specialNames = [ + 'phi-4', 'gemma-2-27b-it', 'Llama-3.2-11B-Vision-Instruct', 'stable-diffusion-3-medium', diff --git a/resources/images/model-player/Phi-4.jpeg b/resources/images/model-player/Phi-4.jpeg new file mode 100644 index 000000000..a84febf07 Binary files /dev/null and b/resources/images/model-player/Phi-4.jpeg differ