diff --git a/requirements.txt b/requirements.txt index b378abf..177a0fe 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,11 +10,11 @@ scipy==1.6.2 setuptools==49.6.0 --find-links https://download.pytorch.org/whl/torch_stable.html -torch==1.10.0+cu111 +torch==1.8.0+cu111 --find-links https://download.pytorch.org/whl/torch_stable.html -torchvision==0.11.0+cu111 +torchvision==0.9.0+cu111 --find-links https://download.pytorch.org/whl/torch_stable.html -torchaudio==0.10.0 +torchaudio==0.8.0 coremltools==5.2.0 onnx==1.7.0