From 019dfc5fa38730bdd61fc30f9d1c9dec03c49d4a Mon Sep 17 00:00:00 2001
From: cuiyaning <cuiyaning@dp.tech>
Date: Mon, 24 Jun 2024 09:24:55 +0000
Subject: [PATCH] add package push

---
 .github/workflows/docker.yml | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
index e4e1475..292418c 100644
--- 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/*
\ No newline at end of file