loosen requirements.txt

Signed-off-by: Oleg S <[email protected]>
instructlab · Jun 20, 2024 · 4ff2b4d · 4ff2b4d
1 parent c6bffd4
commit 4ff2b4d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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