chore: support torch 2.9 uploading wheel to hf (#598)

* chore: support uploading wheel to hf

* support ready_for_review

Author: lmxyy

.github/workflows/pr-test.yaml

@@ -1,7 +1,7 @@
 name: PR Tests
 on:
   pull_request:
-    types: [opened, synchronize, reopened]
+    types: [opened, synchronize, reopened, ready_for_review]
     paths:
       - "nunchaku/**"
       - "src/**"
@@ -98,8 +98,8 @@ jobs:
           comfy node install comfyui-inpainteasy
           cp -r custom_nodes/ComfyUI-nunchaku/tests nunchaku_tests
           pip install -r nunchaku_tests/requirements.txt
-          HF_TOKEN=${{ secrets.HF_TOKEN }} python custom_nodes/ComfyUI-nunchaku/scripts/download_models.py
-          HF_TOKEN=${{ secrets.HF_TOKEN }} python custom_nodes/ComfyUI-nunchaku/scripts/download_test_data.py
+          python custom_nodes/ComfyUI-nunchaku/scripts/download_models.py
+          python custom_nodes/ComfyUI-nunchaku/scripts/download_test_data.py
       - name: Run ComfyUI tests
         run: |
           source $(conda info --base)/etc/profile.d/conda.sh

.github/workflows/release-build.yaml

@@ -51,7 +51,7 @@ jobs:
     strategy:
       matrix:
         python: ["3.10", "3.11", "3.12"]
-        torch: ["2.5", "2.6", "2.7", "2.8"]
+        torch: ["2.5", "2.6", "2.7", "2.8", "2.9"]
     steps:
       - name: Checkout to the tag
         uses: actions/checkout@v4
@@ -82,6 +82,18 @@ jobs:
           prerelease: false
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.12'
+      - name: Upload wheels to HuggingFace Hub
+        env:
+          HF_TOKEN: ${{ secrets.HF_WHEEL_UPLOAD_TOKEN }}
+        run: |
+          pip install huggingface_hub
+          for whl in dist/*.whl; do
+            hf upload nunchaku-tech/nunchaku "$whl" .
+          done
       - name: Clean up
         if: always()
         run: bash scripts/linux_cleanup.sh
@@ -92,7 +104,7 @@ jobs:
     strategy:
       matrix:
         python: ["3.10", "3.11", "3.12"]
-        torch: ["2.5", "2.6", "2.7", "2.8"]
+        torch: ["2.5", "2.6", "2.7", "2.8", "2.9"]
     steps:
       - name: Checkout to the tag
         uses: actions/checkout@v4
@@ -129,3 +141,16 @@ jobs:
           prerelease: false
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.12'
+      - name: Upload wheels to HuggingFace Hub
+        shell: cmd
+        env:
+          HF_TOKEN: ${{ secrets.HF_WHEEL_UPLOAD_TOKEN }}
+        run: |
+          pip install huggingface_hub
+          for %%w in (dist\*.whl) do (
+            hf upload nunchaku-tech/nunchaku "%%w" .
+          )