Commit fef3de6d authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): use torch<2 when exporting to onnx

parent 9e30fe7f
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -38,6 +38,7 @@ jobs:
          pip install --upgrade flake8 setuptools wheel twine
          pip install -r requirements.txt
          pip install -r requirements-test.txt
          pip install 'torch<2'
          pip install -r requirements-zoo.txt
      - name: Prepare dataset
        uses: nick-fields/retry@v2
+1 −1
Original line number Diff line number Diff line
torch<2
torch
lpips
matplotlib
torchvision