diff --git a/.github/workflows/pypi_publish.yaml b/.github/workflows/pypi_publish.yaml
index e2d0fecb2c..1233ae425e 100644
--- a/.github/workflows/pypi_publish.yaml
+++ b/.github/workflows/pypi_publish.yaml
@@ -52,7 +52,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install -U setuptools wheel twine
+          pip install -U setuptools wheel twine torch
       
       - name: Build the dist files
         run: python setup.py bdist_wheel && python setup.py sdist