sta pypi test

hao-ai-lab · Mar 3, 2025 · 65fec08 · 65fec08
1 parent 50a0648
commit 65fec08
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/sta-publish.yml b/.github/workflows/sta-publish.yml
@@ -17,7 +17,7 @@ jobs:
       - name: Checkout code
         uses: actions/checkout@v3
         with:
-          fetch-depth: 1
+          fetch-depth: 2
 
       - name: Check if version changed
         id: check-version
@@ -95,7 +95,7 @@ jobs:
           # e.g. we can have system CUDA version being 11.7 but if torch==1.12 then we need to download the wheel from cu116
           # see https://github.com/pytorch/pytorch/blob/main/RELEASE.md#release-compatibility-matrix
           export TORCH_CUDA_VERSION=124
-          pip install --no-cache-dir torch==${{ matrix.torch-version }} --index-url https://download.pytorch.org/whl/cu${TORCH_CUDA_VERSION}
+          pip install --no-cache-dir torch==2.5.1 --index-url https://download.pytorch.org/whl/cu${TORCH_CUDA_VERSION}
           nvcc --version
           python --version
           python -c "import torch; print('PyTorch:', torch.__version__)"

diff --git a/csrc/sliding_tile_attention/setup.py b/csrc/sliding_tile_attention/setup.py
@@ -9,7 +9,7 @@
 
 # Package metadata
 PACKAGE_NAME = "st_attn"
-VERSION = "0.0.1.dev0"
+VERSION = "0.0.1.dev1"
 AUTHOR = "Hao AI Lab"
 DESCRIPTION = "Sliding Tile Atteniton Kernel Used in FastVideo"
 URL = "https://github.com/hao-ai-lab/FastVideo/tree/main/csrc/sliding_tile_attention"