Publish onediff on pypi automatically

.github/workflows/pub.yml

@@ -0,0 +1,34 @@
+name: Publish
+
+on:
+  push:
+    branches:
+      - "pub/*"
+  schedule:
+    - cron: "0 1 * * *"
+
+concurrency:
+  group: onediff-pub-${{ github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  publish:
+    if: github.repository == 'siliconflow/onediff'
+    runs-on: [ubuntu-latest]
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+      - name: Get current date
+        id: date
+        run: |
+          sed -i '1d' "src/onediff/__init__.py"
+          formatted_date=$(date +'%Y%m%d%H%M')
+          echo "__version__ = \"0.12.1.dev${formatted_date}\"" >> src/onediff/__init__.py
+      - run: cat src/onediff/__init__.py
+      - run: python3 -m pip install build
+      - run: python3 -m build
+      - name: Publish package
+        uses: pypa/gh-action-pypi-publish@27b31702a0e7fc50959f5ad993c78deac1bdfc29
+        with:
+          user: __token__
+          password: ${{ secrets.PYPI_API_TOKEN }}

README.md

@@ -130,6 +130,12 @@ python3 -m pip install "torch" "transformers==4.27.1" "diffusers[torch]==0.19.3"
 ```
 
 #### 3. Install OneDiff
+
+- From PyPI
+```
+python3 -m pip install --pre onediff
+```
+- From source
 ```
 git clone https://github.com/siliconflow/onediff.git
 cd onediff && python3 -m pip install -e .
@@ -154,15 +160,20 @@ python3 -m pip install huggingface_hub
 - bump version in these files:
 
   ```
-  setup.py
+  .github/workflows/pub.yml
   src/onediff/__init__.py
   ```
 
+- install build package
+  ```bash
+  python3 -m pip install build
+  ```
+
 - build wheel
 
-  ```
+  ```bash
   rm -rf dist
-  python3 setup.py bdist_wheel
+  python3 -m build
   ```
 
 - upload to pypi

setup.py

@@ -1,8 +1,18 @@
 from setuptools import find_packages, setup
 
+
+def get_version():
+    variables = {}
+    with open("src/onediff/__init__.py", "r") as f:
+        for line in f:
+            if line.startswith("__version__"):
+                exec(line, variables)
+    return variables["__version__"]
+
+
 setup(
     name="onediff",
-    version="0.12.1.dev",
+    version=get_version(),
     description="an out-of-the-box acceleration library for diffusion models",
     url="https://github.com/siliconflow/onediff",
     author="OneDiff contributors",
@@ -30,4 +40,6 @@
         "Programming Language :: Python :: 3.10",
         "Topic :: Scientific/Engineering :: Artificial Intelligence",
     ],
+    long_description=open("README.md").read(),
+    long_description_content_type="text/markdown",
 )

src/onediff/__init__.py

@@ -1,3 +1,3 @@
-__version__ = "0.12.1.dev"
+__version__ = "0.12.1.dev1"
 __author__ = "OneDiff"
 __credits__ = "OneDiff contributors"