add package push

deepmodeling · Jun 24, 2024 · 019dfc5 · 019dfc5
1 parent 1fe443e
commit 019dfc5
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
@@ -53,3 +53,28 @@ jobs:
           context: ./unimol/docker/
           push: true
           tags: dptechnology/unimol:latest-pytorch1.11.0-cuda11.3
+
+  publish_package:
+    name: Publish package
+    needs: [docker]
+
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+
+      - uses: actions/setup-python@v4
+        with:
+          python-version: '3.10'
+
+      - name: Build core package
+        env:
+          FLASH_ATTENTION_SKIP_CUDA_BUILD: "TRUE"
+        run: |
+          python setup.py sdist --dist-dir=dist
+
+      - name: Deploy
+        env:
+          TWINE_USERNAME: "__token__"
+          TWINE_PASSWORD: ${{ secrets.PYPI_TOKEN }}
+        run: |
+          python -m twine upload dist/*