diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index eb1e74bc01..ed1443e30a 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..977bf2c375 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -406,6 +406,13 @@ 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, + }, moshi: { prettyLabel: "Moshi", repoName: "Moshi",