FEAT(local-apps): add ComfyUI LocalApp

Signed-off-by: tildebyte <[email protected]>
huggingface · Oct 5, 2024 · 5ebe7a6 · 5ebe7a6
1 parent 4e7a89a
commit 5ebe7a6
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/packages/tasks/src/local-apps.ts b/packages/tasks/src/local-apps.ts
@@ -322,6 +322,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<string, LocalApp>;
 
 export type LocalAppKey = keyof typeof LOCAL_APPS;