upd

zhyncs · Nov 17, 2024 · d5cf950 · d5cf950
1 parent cb1288c
commit d5cf950
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/.github/workflows/cuda12.1-whl-release.yml b/.github/workflows/cuda12.1-whl-release.yml
@@ -2,6 +2,13 @@ name: cuda12.1-whl-release
 
 on:
   workflow_dispatch:
+    inputs:
+      repo:
+        description: "repo"
+        default: "InternLM/lmdeploy"
+      branch:
+        description: "branch"
+        default: "main"
   schedule:
     - cron: '30 1 * * *'
 
@@ -34,7 +41,7 @@ jobs:
           large-packages: true
           swap-storage: false
       - name: Checkout repository
-        run: git clone --depth=1 https://github.com/InternLM/lmdeploy
+        run: git clone -b ${{ inputs.branch }} --depth=1 https://github.com/${{ inputs.repo }}
       - name: Build
         run: |
           echo ${PYTHON_VERSION}
@@ -74,7 +81,7 @@ jobs:
             INPUT_CUDA_VERSION: '12.1.0'
       - name: Build wheel
         run: |
-          git clone --depth=1 https://github.com/InternLM/lmdeploy
+          git clone -b ${{ inputs.branch }} --depth=1 https://github.com/${{ inputs.repo }}
           pushd lmdeploy
           mkdir build
           cd build
@@ -104,7 +111,7 @@ jobs:
       - windows-build
     steps:
       - name: Checkout repository
-        run: git clone --depth=1 https://github.com/InternLM/lmdeploy
+        run: git clone -b ${{ inputs.branch }} --depth=1 https://github.com/${{ inputs.repo }}
       - name: Download artifacts
         uses: actions/download-artifact@v3
       - name: Add cuda version to package name