Skip to content

Commit cc53d2b

Browse files
committed
upgrade transformers and torch
Signed-off-by: Sukriti-Sharma4 <[email protected]>
1 parent f1fd130 commit cc53d2b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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.1",
32-
"transformers>=4.45,<4.46",
33-
"torch>=2.2.0,<2.5",
32+
"transformers @ git+https://github.com/mayank31398/transformers@0032dc796b0f7052ab095cd2cdcbfa8e36a75279",
33+
"torch>=2.2.0,<=2.6",
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)