diff --git a/bench/requirements.txt b/bench/requirements.txt index 6d8870e..79efce3 100644 --- a/bench/requirements.txt +++ b/bench/requirements.txt @@ -3,4 +3,4 @@ # pip install torch==2.4.1 --index-url https://download.pytorch.org/whl/cpu numpy>=1.24 sentence-transformers==3.1.1 -transformers==4.44.2 +transformers==4.57.6 diff --git a/pyproject.toml b/pyproject.toml index 26bd351..23b0df9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -55,7 +55,7 @@ dependencies = ["numpy>=1.24"] embeddings = [ "sentence-transformers>=2.6", "numpy<2; sys_platform == 'darwin' and platform_machine == 'x86_64'", - "transformers<4.50; sys_platform == 'darwin' and platform_machine == 'x86_64'", + "transformers<4.58; sys_platform == 'darwin' and platform_machine == 'x86_64'", "sentence-transformers<5; sys_platform == 'darwin' and platform_machine == 'x86_64'", ]