diff --git a/requirements.txt b/requirements.txt index 8bb8d804..1bbec7e4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,13 +4,13 @@ py-cpuinfo # we set this to be above 0a0 so that it doesn't # replace custom pytorch images with the 2.3.0 torch>=2.3.0a0 -transformers==4.41.2 +transformers>=4.41.2 datasets>=2.15.0 numba numpy rich dolomite-engine @ git+https://github.com/ibm-granite/dolomite-engine.git@main -trl==0.9.4 +trl>=0.9.4 peft pydantic>=2.7.0