fix publish to pypi (#439)

open-mmlab · Sep 14, 2022 · f6551e1 · f6551e1
1 parent 6995465
commit f6551e1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/publish-to-pypi.yml b/.github/workflows/publish-to-pypi.yml
@@ -18,6 +18,7 @@ jobs:
           python-version: 3.7
       - name: Build MMGeneration
         run: |
+          pip install torch==1.8.1+cpu torchvision==0.9.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
           pip install wheel
           python setup.py sdist
       - name: Publish distribution to PyPI