Adds GPT4All as a local model option

huggingface · Aug 14, 2024 · 2b92175 · 2b92175
1 parent 33e6316
commit 2b92175
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/packages/tasks/src/local-apps.ts b/packages/tasks/src/local-apps.ts
@@ -250,6 +250,13 @@ 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}`),
 	},
+	gpt4all: {
+		prettyLabel: "GPT4All",
+		docsUrl: "https://nomic.ai/gpt4all",
+		mainTask: "text-generation",
+		displayOnModelPage: isGgufModel,
+		deeplink: (model) => new URL(`gpt4all://models/huggingface/${model.id}`),
+	},
 } satisfies Record<string, LocalApp>;
 
 export type LocalAppKey = keyof typeof LOCAL_APPS;