diff --git a/.github/workflows/pub.yml b/.github/workflows/pub.yml
index 3931247e3..7bd6d4efe 100644
--- a/.github/workflows/pub.yml
+++ b/.github/workflows/pub.yml
@@ -4,13 +4,15 @@ on:
   push:
     branches:
       - "pub/*"
-  schedule:
-    - cron: "0 1 * * *"
+      - main
 
 concurrency:
   group: onediff-pub-${{ github.ref }}
   cancel-in-progress: true
 
+env:
+  VERSION_LOCAL_SCHEME: no-local-version
+
 jobs:
   publish:
     if: github.repository == 'siliconflow/onediff'
@@ -22,10 +24,7 @@ jobs:
           fetch-depth: 0
       - run: cat src/onediff/__init__.py
       - run: python3 -m pip install build
-      - name: Set environment variable
-        env:
-          VERSION_LOCAL_SCHEME: no-local-version
-        run: echo "VERSION_LOCAL_SCHEME=$VERSION_LOCAL_SCHEME"
+      - run: echo "VERSION_LOCAL_SCHEME=$VERSION_LOCAL_SCHEME"
       - run: python3 -m build
       - name: Publish package
         uses: pypa/gh-action-pypi-publish@27b31702a0e7fc50959f5ad993c78deac1bdfc29
diff --git a/onediff_diffusers_extensions/setup.py b/onediff_diffusers_extensions/setup.py
index 36a4c4a68..553647876 100644
--- a/onediff_diffusers_extensions/setup.py
+++ b/onediff_diffusers_extensions/setup.py
@@ -4,6 +4,7 @@
 
 
 local_scheme = os.getenv("VERSION_LOCAL_SCHEME", "node-and-date")
+
 setup(
     name="onediffx",
     description="onediff extensions for diffusers",
@@ -38,7 +39,7 @@
     long_description_content_type="text/markdown",
     use_scm_version={
         "root": "../",
-        "write_to": "./onediff_diffusers_extensions/onediffx/_version.py",
+        "version_file": "./onediffx/_version.py",
         "fallback_version": "0.0.0",
         "version_scheme": "guess-next-dev",
         "local_scheme": local_scheme,
diff --git a/setup.py b/setup.py
index 8c4fd5a68..7fc6d7148 100644
--- a/setup.py
+++ b/setup.py
@@ -44,7 +44,7 @@
         ],
     },
     use_scm_version={
-        "write_to": "src/onediff/_version.py",
+        "version_file": "src/onediff/_version.py",
         "fallback_version": "0.0.0",
         "version_scheme": "guess-next-dev",
         "local_scheme": local_scheme,