Skip to content

Commit a977bc4

Browse files
mergennachinpsiddh
authored andcommitted
Add torch, torchvision and torchaudio dependencies
1 parent 12edd72 commit a977bc4

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

pyproject.toml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,9 @@ dependencies=[
6767
"pyyaml",
6868
"ruamel.yaml",
6969
"sympy",
70+
"torch>=2.8.0,<2.9.0",
71+
"torchaudio>=2.8.0,<2.9.0",
72+
"torchvision>=0.23.0,<0.24.0",
7073
"torchao==0.12.0",
7174
"tabulate",
7275
# See also third-party/TARGETS for buck's typing-extensions version.

0 commit comments

Comments
 (0)