Skip to content

Commit 5ebe7a6

Browse files
committed
FEAT(local-apps): add ComfyUI LocalApp
Signed-off-by: tildebyte <[email protected]>
1 parent 4e7a89a commit 5ebe7a6

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

packages/tasks/src/local-apps.ts

+9
Original file line numberDiff line numberDiff line change
@@ -322,6 +322,15 @@ export const LOCAL_APPS = {
322322
displayOnModelPage: (model) => model.library_name === "diffusers" && model.pipeline_tag === "text-to-image",
323323
deeplink: (model) => new URL(`https://models.invoke.ai/huggingface/${model.id}`),
324324
},
325+
comfyui: {
326+
prettyLabel: "ComfyUI",
327+
docsUrl: "https://docs.comfy.org/get_started/introduction",
328+
mainTask: "text-to-image",
329+
displayOnModelPage: (model: ModelData) =>
330+
(isGgufModel(model) || (isTransformersModel(model) && model.tags.includes("safetensors"))) &&
331+
model.tags.includes("text-to-image"),
332+
deeplink: (model) => new URL(`https://huggingface.co/${model.id}`),
333+
},
325334
} satisfies Record<string, LocalApp>;
326335

327336
export type LocalAppKey = keyof typeof LOCAL_APPS;

0 commit comments

Comments
 (0)