Skip to content

Commit 40cc0df

Browse files
committed
fix: Upgrade transformers, torch and accelerate packages version
Signed-off-by: Akash-Nayak <akash19nayak@gmail.com>
1 parent eb72480 commit 40cc0df

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,8 @@ classifiers=[
2929
dependencies = [
3030
"numpy>=1.26.4,<2.0",
3131
"accelerate>=0.20.3,!=0.34,<1.7",
32-
"transformers>=4.52.3",
33-
"torch==2.6.0",
32+
"transformers>=4.49,<=4.52.4",
33+
"torch>=2.2.0,<2.7",
3434
"sentencepiece>=0.1.99,<0.3",
3535
"tokenizers>=0.13.3,<1.0",
3636
"tqdm>=4.66.2,<5.0",

0 commit comments

Comments
 (0)