From 4c737781ed5e41f976eaa9dd2aaa163b739d0970 Mon Sep 17 00:00:00 2001 From: Tom Aarsen Date: Thu, 26 Sep 2024 16:42:23 +0200 Subject: [PATCH 1/3] Add the Model2Vec library --- packages/tasks/src/model-libraries-snippets.ts | 6 ++++++ packages/tasks/src/model-libraries.ts | 8 ++++++++ 2 files changed, 14 insertions(+) diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index eb1e74bc01..b2a73cacc6 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -951,6 +951,12 @@ export const mlxim = (model: ModelData): string[] => [ model = create_model(${model.id})`, ]; +export const model2vec = (model: ModelData): string[] => [ + `from model2vec import StaticModel + +model = StaticModel.from_pretrained("${model.id}")`, +] + export const nemo = (model: ModelData): string[] => { let command: string[] | undefined = undefined; // Resolve the tag to a nemo domain/sub-domain diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 7ebde8b6dc..3eec600af2 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -406,6 +406,14 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { filter: false, countDownloads: `path:"mlc-chat-config.json"`, }, + model2vec: { + prettyLabel: "Model2Vec", + repoName: "model2vec", + repoUrl: "https://github.com/MinishLab/model2vec", + snippets: snippets.model2vec, + filter: false, + countDownloads: `path_extension:"safetensors"`, + }, moshi: { prettyLabel: "Moshi", repoName: "Moshi", From 1312a230585c7f4221d264cba96f83b5a405c9c6 Mon Sep 17 00:00:00 2001 From: Tom Aarsen Date: Thu, 26 Sep 2024 16:50:53 +0200 Subject: [PATCH 2/3] Run pnpm format --- packages/tasks/src/model-libraries-snippets.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index b2a73cacc6..ed1443e30a 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -955,7 +955,7 @@ export const model2vec = (model: ModelData): string[] => [ `from model2vec import StaticModel model = StaticModel.from_pretrained("${model.id}")`, -] +]; export const nemo = (model: ModelData): string[] => { let command: string[] | undefined = undefined; From 960ef9998943a815ec0e58677ffe6a5b3ecab897 Mon Sep 17 00:00:00 2001 From: Tom Aarsen Date: Thu, 26 Sep 2024 17:46:25 +0200 Subject: [PATCH 3/3] Remove countDownloads as the default config.json should suffice --- packages/tasks/src/model-libraries.ts | 1 - 1 file changed, 1 deletion(-) diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 3eec600af2..977bf2c375 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -412,7 +412,6 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { repoUrl: "https://github.com/MinishLab/model2vec", snippets: snippets.model2vec, filter: false, - countDownloads: `path_extension:"safetensors"`, }, moshi: { prettyLabel: "Moshi",