diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index 9d7313faec..3f387a6915 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -2024,23 +2024,4 @@ torchaudio.save("sample.wav", audio, model.autoencoder.sampling_rate) `, ]; -export const mistral_common = (model: ModelData): string[] => [ - `# We recommend to use vLLM to serve Mistral AI models. -pip install vllm - -# Make sure to have installed the latest version of mistral-common. -pip install --upgrade mistral-common[image,audio] - -# Serve the model with an OpenAI-compatible API. -vllm serve ${model.id} --tokenizer_mode mistral --config_format mistral --load_format mistral --tool-call-parser mistral --enable-auto-tool-choice - -# Query the model with curl in a separate terminal. -curl http://localhost:8000/v1/chat/completions \ - -H "Content-Type: application/json" \ - -d '{ - "model": "${model.id}", - "messages": [{"role": "user", "content": "What is the capital of France?"}] - }'`, -]; - //#endregion diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 9e24cfce46..f99284b63e 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -32,7 +32,7 @@ export interface LibraryUiElement { * Elastic query used to count this library's model downloads * * By default, those files are counted: - * "config.json", "config.yaml", "hyperparams.yaml", "meta.yaml" + * "config.json", "config.yaml", "hyperparams.yaml", "params.json", "meta.yaml" */ countDownloads?: ElasticSearchQuery; /** @@ -631,14 +631,6 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { filter: false, countDownloads: `path_extension:"ckpt"`, }, - mistral_common: { - prettyLabel: "mistral-common", - repoName: "mistral-common", - repoUrl: "https://github.com/mistralai/mistral-common", - docsUrl: "https://mistralai.github.io/mistral-common/", - snippets: snippets.mistral_common, - countDownloads: `path:"config.json" OR path:"params.json"`, - }, mitie: { prettyLabel: "MITIE", repoName: "MITIE",