diff --git a/README.md b/README.md index dc6e6ba67..21a76a744 100644 --- a/README.md +++ b/README.md @@ -44,6 +44,12 @@ pip install flash-attn==2.7.0.post2 To try Sliding Tile Attention (optional), please follow the instruction in [csrc/sliding_tile_attention/README.md](csrc/sliding_tile_attention/README.md) to install STA. +You can also install the Sliding Tile Attention package using + +``` +pip install st_attn==0.0.3 +``` + ## 🚀 Inference ### Inference StepVideo with Sliding Tile Attention First, download the model: diff --git a/csrc/sliding_tile_attention/setup.py b/csrc/sliding_tile_attention/setup.py index 2bd181f9d..d421af691 100644 --- a/csrc/sliding_tile_attention/setup.py +++ b/csrc/sliding_tile_attention/setup.py @@ -9,7 +9,7 @@ target = target.lower() # Package metadata PACKAGE_NAME = "st_attn" -VERSION = "0.0.2" +VERSION = "0.0.3" 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"