diff --git a/pyproject.toml b/pyproject.toml index b21d605..3a25155 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -18,7 +18,8 @@ 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", ] # On a mac, install optional dependencies with `pip install '.[dev]'` (include the single quotes) @@ -49,7 +50,7 @@ pipelines = [ "nltk", "sentence-transformers>=2.3.1", "peft>=0.8.2", - "trl>=0.7.10", + "trl==0.7.10", "datasets", "tensorboardX", ]