diff --git a/docs/examples/Supported_Models.ipynb b/docs/examples/Supported_Models.ipynb index d20ad9f12..935d2808b 100644 --- a/docs/examples/Supported_Models.ipynb +++ b/docs/examples/Supported_Models.ipynb @@ -2,36 +2,41 @@ "cells": [ { "cell_type": "code", + "execution_count": 1, "metadata": { "ExecuteTime": { "end_time": "2024-11-13T09:01:03.324551Z", "start_time": "2024-11-13T09:01:03.234711Z" + }, + "execution": { + "iopub.execute_input": "2026-07-22T21:47:58.683042Z", + "iopub.status.busy": "2026-07-22T21:47:58.682929Z", + "iopub.status.idle": "2026-07-22T21:47:58.756340Z", + "shell.execute_reply": "2026-07-22T21:47:58.755741Z" } }, + "outputs": [], "source": [ "%load_ext autoreload\n", "%autoreload 2" - ], - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "The autoreload extension is already loaded. To reload it, use:\n", - " %reload_ext autoreload\n" - ] - } - ], - "execution_count": 10 + ] }, { "cell_type": "code", + "execution_count": 2, "metadata": { "ExecuteTime": { "end_time": "2024-11-13T09:01:04.505772Z", "start_time": "2024-11-13T09:01:04.493296Z" + }, + "execution": { + "iopub.execute_input": "2026-07-22T21:47:58.757895Z", + "iopub.status.busy": "2026-07-22T21:47:58.757774Z", + "iopub.status.idle": "2026-07-22T21:48:10.438389Z", + "shell.execute_reply": "2026-07-22T21:48:10.437700Z" } }, + "outputs": [], "source": [ "import pandas as pd\n", "\n", @@ -41,10 +46,10 @@ " LateInteractionTextEmbedding,\n", " ImageEmbedding,\n", ")\n", - "from fastembed.rerank.cross_encoder import TextCrossEncoder" - ], - "outputs": [], - "execution_count": 11 + "from fastembed.rerank.cross_encoder import TextCrossEncoder\n", + "\n", + "pd.set_option(\"display.max_colwidth\", None)" + ] }, { "cell_type": "markdown", @@ -55,79 +60,22 @@ }, { "cell_type": "code", + "execution_count": 3, "metadata": { "ExecuteTime": { "end_time": "2024-11-13T09:01:05.812271Z", "start_time": "2024-11-13T09:01:05.795846Z" + }, + "execution": { + "iopub.execute_input": "2026-07-22T21:48:10.440580Z", + "iopub.status.busy": "2026-07-22T21:48:10.440336Z", + "iopub.status.idle": "2026-07-22T21:48:10.482086Z", + "shell.execute_reply": "2026-07-22T21:48:10.481626Z" } }, - "source": [ - "supported_models = (\n", - " pd.DataFrame(TextEmbedding.list_supported_models())\n", - " .sort_values(\"size_in_GB\")\n", - " .drop(columns=[\"sources\", \"model_file\", \"additional_files\"])\n", - " .reset_index(drop=True)\n", - ")\n", - "supported_models" - ], "outputs": [ { "data": { - "text/plain": [ - " model dim \\\n", - "0 BAAI/bge-small-en-v1.5 384 \n", - "1 BAAI/bge-small-zh-v1.5 512 \n", - "2 snowflake/snowflake-arctic-embed-xs 384 \n", - "3 sentence-transformers/all-MiniLM-L6-v2 384 \n", - "4 jinaai/jina-embeddings-v2-small-en 512 \n", - "5 BAAI/bge-small-en 384 \n", - "6 snowflake/snowflake-arctic-embed-s 384 \n", - "7 nomic-ai/nomic-embed-text-v1.5-Q 768 \n", - "8 BAAI/bge-base-en-v1.5 768 \n", - "9 sentence-transformers/paraphrase-multilingual-... 384 \n", - "10 Qdrant/clip-ViT-B-32-text 512 \n", - "11 jinaai/jina-embeddings-v2-base-de 768 \n", - "12 BAAI/bge-base-en 768 \n", - "13 snowflake/snowflake-arctic-embed-m 768 \n", - "14 nomic-ai/nomic-embed-text-v1.5 768 \n", - "15 jinaai/jina-embeddings-v2-base-en 768 \n", - "16 nomic-ai/nomic-embed-text-v1 768 \n", - "17 snowflake/snowflake-arctic-embed-m-long 768 \n", - "18 mixedbread-ai/mxbai-embed-large-v1 1024 \n", - "19 jinaai/jina-embeddings-v2-base-code 768 \n", - "20 sentence-transformers/paraphrase-multilingual-... 768 \n", - "21 snowflake/snowflake-arctic-embed-l 1024 \n", - "22 thenlper/gte-large 1024 \n", - "23 BAAI/bge-large-en-v1.5 1024 \n", - "24 intfloat/multilingual-e5-large 1024 \n", - "\n", - " description license size_in_GB \n", - "0 Text embeddings, Unimodal (text), English, 512... mit 0.067 \n", - "1 Text embeddings, Unimodal (text), Chinese, 512... mit 0.090 \n", - "2 Text embeddings, Unimodal (text), English, 512... apache-2.0 0.090 \n", - "3 Text embeddings, Unimodal (text), English, 256... apache-2.0 0.090 \n", - "4 Text embeddings, Unimodal (text), English, 819... apache-2.0 0.120 \n", - "5 Text embeddings, Unimodal (text), English, 512... mit 0.130 \n", - "6 Text embeddings, Unimodal (text), English, 512... apache-2.0 0.130 \n", - "7 Text embeddings, Multimodal (text, image), Eng... apache-2.0 0.130 \n", - "8 Text embeddings, Unimodal (text), English, 512... mit 0.210 \n", - "9 Text embeddings, Unimodal (text), Multilingual... apache-2.0 0.220 \n", - "10 Text embeddings, Multimodal (text&image), Engl... mit 0.250 \n", - "11 Text embeddings, Unimodal (text), Multilingual... apache-2.0 0.320 \n", - "12 Text embeddings, Unimodal (text), English, 512... mit 0.420 \n", - "13 Text embeddings, Unimodal (text), English, 512... apache-2.0 0.430 \n", - "14 Text embeddings, Multimodal (text, image), Eng... apache-2.0 0.520 \n", - "15 Text embeddings, Unimodal (text), English, 819... apache-2.0 0.520 \n", - "16 Text embeddings, Multimodal (text, image), Eng... apache-2.0 0.520 \n", - "17 Text embeddings, Unimodal (text), English, 204... apache-2.0 0.540 \n", - "18 Text embeddings, Unimodal (text), English, 512... apache-2.0 0.640 \n", - "19 Text embeddings, Unimodal (text), Multilingual... apache-2.0 0.640 \n", - "20 Text embeddings, Unimodal (text), Multilingual... apache-2.0 1.000 \n", - "21 Text embeddings, Unimodal (text), English, 512... apache-2.0 1.020 \n", - "22 Text embeddings, Unimodal (text), English, 512... mit 1.200 \n", - "23 Text embeddings, Unimodal (text), English, 512... mit 1.200 \n", - "24 Text embeddings, Unimodal (text), Multilingual... mit 2.240 " - ], "text/html": [ "
\n", "