diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml index dc82d517..9a74ed54 100644 --- a/.github/workflows/python-publish.yml +++ b/.github/workflows/python-publish.yml @@ -23,7 +23,7 @@ jobs: python -m pip install --upgrade pip pip install setuptools wheel twine pip install -r requirements.txt - pip install torch torchvision + pip install torch==1.10.0+cpu torchvision==0.11.1+cpu torchaudio==0.10.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html - name: Build and publish env: TWINE_USERNAME: ${{ secrets.PYPI_USERNAME }}