Skip to content

Commit

Permalink
loosen requirements.txt
Browse files Browse the repository at this point in the history
Signed-off-by: Oleg S <97077423+RobotSail@users.noreply.github.com>
  • Loading branch information
RobotSail committed Jun 20, 2024
1 parent c6bffd4 commit cb6bb4c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit cb6bb4c

Please sign in to comment.