diff --git a/dashboard/src/routes/+page.svelte b/dashboard/src/routes/+page.svelte index 9edadb62..183b6547 100644 --- a/dashboard/src/routes/+page.svelte +++ b/dashboard/src/routes/+page.svelte @@ -130,6 +130,15 @@ model.tasks.includes("ImageToImage") ); } + + // Helper to check if a model supports image editing + function modelSupportsImageEditing(modelId: string): boolean { + const model = models.find( + (m) => m.id === modelId || m.hugging_face_id === modelId, + ); + if (!model?.tasks) return false; + return model.tasks.includes("ImageToImage"); + } let selectedSharding = $state<"Pipeline" | "Tensor">("Pipeline"); type InstanceMeta = "MlxRing" | "MlxIbv" | "MlxJaccl"; @@ -2369,6 +2378,9 @@ {@const isImageModel = modelSupportsImageGeneration( foundModel.id, )} + {@const isImageEditModel = modelSupportsImageEditing( + foundModel.id, + )} @@ -2395,6 +2407,22 @@ {/if} + {#if isImageEditModel} + + + + + {/if} {foundModel.name || foundModel.id} @@ -2467,6 +2495,9 @@ {@const isImageModel = modelSupportsImageGeneration( model.id, )} + {@const isImageEditModel = modelSupportsImageEditing( + model.id, + )}