From 15ad82bb6aff4c45900781944ad6ad20ca01a24f Mon Sep 17 00:00:00 2001 From: Carlos Garcia Jurado Suarez Date: Tue, 11 Mar 2025 16:55:06 -0700 Subject: [PATCH 1/3] fix: Bound numpy version to avoid incompatibilities --- pyproject.toml | 1 + 1 file changed, 1 insertion(+) diff --git a/pyproject.toml b/pyproject.toml index b21d605..55f4e4b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -19,6 +19,7 @@ dependencies = [ # This works, while installing from pytorch and cuda from conda does not", "torch==2.0.1", "transformers>=4.37.2", + "numpy<2.0", ] # On a mac, install optional dependencies with `pip install '.[dev]'` (include the single quotes) From 6837447405052e1e1c3603023219b646e5e16557 Mon Sep 17 00:00:00 2001 From: Carlos Garcia Jurado Suarez Date: Tue, 11 Mar 2025 17:10:09 -0700 Subject: [PATCH 2/3] fix: pin trl version to 0.7.10 --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 55f4e4b..92dd320 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -50,7 +50,7 @@ pipelines = [ "nltk", "sentence-transformers>=2.3.1", "peft>=0.8.2", - "trl>=0.7.10", + "trl==0.7.10", "datasets", "tensorboardX", ] From 08fc2a468ada62d0eb6ae9425acdf5a2df4e87aa Mon Sep 17 00:00:00 2001 From: Carlos Garcia Jurado Suarez Date: Tue, 11 Mar 2025 19:09:04 -0700 Subject: [PATCH 3/3] fix: pin transformers version --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 92dd320..3a25155 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -18,7 +18,7 @@ dependencies = [ "scipy", # This works, while installing from pytorch and cuda from conda does not", "torch==2.0.1", - "transformers>=4.37.2", + "transformers==4.37.2", "numpy<2.0", ]