From 1484406a8d274fdfae5a386003e23efaaf9e98a4 Mon Sep 17 00:00:00 2001 From: Andy Lee Date: Sun, 5 Oct 2025 19:01:58 -0700 Subject: [PATCH] chore: align core deps with transformers pin --- packages/leann-core/pyproject.toml | 6 +++--- uv.lock | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/packages/leann-core/pyproject.toml b/packages/leann-core/pyproject.toml index 71f6bb7..6d47bd7 100644 --- a/packages/leann-core/pyproject.toml +++ b/packages/leann-core/pyproject.toml @@ -18,14 +18,14 @@ dependencies = [ "pyzmq>=23.0.0", "msgpack>=1.0.0", "torch>=2.0.0", - "sentence-transformers>=2.2.0", + "sentence-transformers>=2.2.0,<3.0", "llama-index-core>=0.12.0", "llama-index-readers-file>=0.4.0", # Essential for document reading "llama-index-embeddings-huggingface>=0.5.5", # For embeddings "python-dotenv>=1.0.0", "openai>=1.0.0", "huggingface-hub>=0.20.0", - "transformers>=4.30.0", + "transformers>=4.30.0,<4.43", "requests>=2.25.0", "accelerate>=0.20.0", "PyPDF2>=3.0.0", @@ -40,7 +40,7 @@ dependencies = [ [project.optional-dependencies] colab = [ "torch>=2.0.0,<3.0.0", # Limit torch version to avoid conflicts - "transformers>=4.30.0,<5.0.0", # Limit transformers version + "transformers>=4.30.0,<4.43", # Limit transformers version for Python 3.9 compatibility "accelerate>=0.20.0,<1.0.0", # Limit accelerate version ] diff --git a/uv.lock b/uv.lock index d212af5..7298644 100644 --- a/uv.lock +++ b/uv.lock @@ -2241,12 +2241,12 @@ requires-dist = [ { name = "python-dotenv", specifier = ">=1.0.0" }, { name = "pyzmq", specifier = ">=23.0.0" }, { name = "requests", specifier = ">=2.25.0" }, - { name = "sentence-transformers", specifier = ">=2.2.0" }, + { name = "sentence-transformers", specifier = ">=2.2.0,<3.0" }, { name = "torch", specifier = ">=2.0.0" }, { name = "torch", marker = "extra == 'colab'", specifier = ">=2.0.0,<3.0.0" }, { name = "tqdm", specifier = ">=4.60.0" }, - { name = "transformers", specifier = ">=4.30.0" }, - { name = "transformers", marker = "extra == 'colab'", specifier = ">=4.30.0,<5.0.0" }, + { name = "transformers", specifier = ">=4.30.0,<4.43" }, + { name = "transformers", marker = "extra == 'colab'", specifier = ">=4.30.0,<4.43" }, ] provides-extras = ["colab"]