From 12238a7a0a066b5b352857c2481ed417047b6631 Mon Sep 17 00:00:00 2001 From: wilyJ80 Date: Wed, 4 Mar 2026 11:31:22 -0300 Subject: [PATCH 1/3] fix: pandas broken packages --- docs/examples/Supported_Models.ipynb | 373 +++--- poetry.lock | 1654 +++++++++++++------------- pyproject.toml | 5 +- 3 files changed, 995 insertions(+), 1037 deletions(-) diff --git a/docs/examples/Supported_Models.ipynb b/docs/examples/Supported_Models.ipynb index d20ad9f12..1ffbaf81a 100644 --- a/docs/examples/Supported_Models.ipynb +++ b/docs/examples/Supported_Models.ipynb @@ -2,16 +2,13 @@ "cells": [ { "cell_type": "code", + "execution_count": 5, "metadata": { "ExecuteTime": { "end_time": "2024-11-13T09:01:03.324551Z", "start_time": "2024-11-13T09:01:03.234711Z" } }, - "source": [ - "%load_ext autoreload\n", - "%autoreload 2" - ], "outputs": [ { "name": "stdout", @@ -22,16 +19,21 @@ ] } ], - "execution_count": 10 + "source": [ + "%load_ext autoreload\n", + "%autoreload 2" + ] }, { "cell_type": "code", + "execution_count": 2, "metadata": { "ExecuteTime": { "end_time": "2024-11-13T09:01:04.505772Z", "start_time": "2024-11-13T09:01:04.493296Z" } }, + "outputs": [], "source": [ "import pandas as pd\n", "\n", @@ -42,9 +44,7 @@ " ImageEmbedding,\n", ")\n", "from fastembed.rerank.cross_encoder import TextCrossEncoder" - ], - "outputs": [], - "execution_count": 11 + ] }, { "cell_type": "markdown", @@ -55,79 +55,16 @@ }, { "cell_type": "code", + "execution_count": 12, "metadata": { "ExecuteTime": { "end_time": "2024-11-13T09:01:05.812271Z", "start_time": "2024-11-13T09:01:05.795846Z" } }, - "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", "