Skip to content

Commit d6dcfff

Browse files
committed
fix reqs
1 parent 95f8342 commit d6dcfff

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

requirements-test.txt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,7 @@
11
# PyPI doesn't like direct dependencies - see https://github.com/microsoft/microxcaling/issues/22
22

33
mx @ git+https://github.com/microsoft/microxcaling
4+
numpy<2.0.0 # microscaling needs pytorch 2.1, which needs to be less than 2.0
5+
torch==2.1.0 --index-url https://download.pytorch.org/whl/cpu
6+
torchvision==0.16.0
7+
torchaudio==2.1.0

0 commit comments

Comments
 (0)