File tree Expand file tree Collapse file tree 1 file changed +0
-9
lines changed
Expand file tree Collapse file tree 1 file changed +0
-9
lines changed Original file line number Diff line number Diff line change 1818 torch28: torchvision~=0.23.0
1919 torch29: torchvision~=0.24.0
2020
21- # Build onnxsim from sdists for Python 3.12 until http://github.com/daquexian/onnx-simplifier/pull/353
22- py312: onnxsim
23-
2421 # Install megatron-core for special unit tests
2522 megatron-core
2623
4239 # Make sure torch 2.9 is used
4340 torchvision~=0.24.0
4441
45- # Build onnxsim from sdists for Python 3.12 until http://github.com/daquexian/onnx-simplifier/pull/353
46- py312: onnxsim
47-
4842 # ONNX unit tests heavily rely on torch / torchvision
4943 onnx: .[onnx,dev-test]
5044 onnx: torchvision
@@ -80,9 +74,6 @@ commands_pre =
8074 # Install Eagle-3 test dependencies
8175 pip install tiktoken blobfile sentencepiece
8276
83- # Build onnxsim from sdists for Python 3.12 until http://github.com/daquexian/onnx-simplifier/pull/353
84- py312: pip install onnxsim
85-
8677 # NOTE: User is expected to have correct torch-cuda version pre-installed if using --current-env
8778 # to avoid possible CUDA version mismatch
8879 pip install -e .[all,dev-test]
You can’t perform that action at this time.
0 commit comments