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,
+ )}