[STA] Sta release 0.0.3 (#344)

This commit is contained in:
William Lin
2025-04-15 13:06:03 -07:00
committed by GitHub
parent 2325adffa2
commit 67da56628b
2 changed files with 7 additions and 1 deletions
+6
View File
@@ -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:
+1 -1
View File
@@ -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"