From 7a47dc252492239af43d1a2c0a53fc2a7a553600 Mon Sep 17 00:00:00 2001 From: tildebyte <337875+tildebyte@users.noreply.github.com> Date: Tue, 1 Oct 2024 16:16:29 -0400 Subject: [PATCH] FEAT(local-apps): add ComfyUI LocalApp Signed-off-by: tildebyte --- packages/tasks/src/local-apps.ts | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/packages/tasks/src/local-apps.ts b/packages/tasks/src/local-apps.ts index 6ebde2592..34158323a 100644 --- a/packages/tasks/src/local-apps.ts +++ b/packages/tasks/src/local-apps.ts @@ -323,6 +323,15 @@ export const LOCAL_APPS = { displayOnModelPage: (model) => model.library_name === "diffusers" && model.pipeline_tag === "text-to-image", deeplink: (model) => new URL(`https://models.invoke.ai/huggingface/${model.id}`), }, + comfyui: { + prettyLabel: "ComfyUI", + docsUrl: "https://docs.comfy.org/get_started/introduction", + mainTask: "text-to-image", + displayOnModelPage: (model: ModelData) => + (isGgufModel(model) || (isTransformersModel(model) && model.tags.includes("safetensors"))) && + model.tags.includes("text-to-image"), + deeplink: (model) => new URL(`https://huggingface.co/${model.id}`), + }, } satisfies Record; export type LocalAppKey = keyof typeof LOCAL_APPS;