nodes/hao-ai-lab-FastVideo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CI] Add STA-inference/VSA-training test (#527)

Browse Source 
Co-authored-by: Will Lin <wlsaidhi@gmail.com>
This commit is contained in:
Yongqi Chen
2025-06-17 21:13:06 -07:00
committed by GitHub
co-authored by Will Lin
parent de65bec2b7
commit 55074bda76
8 changed files with 254 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile.python3.12
+2 -2
View File
@@ -58,14 +58,14 @@ RUN source $HOME/.local/bin/env && \
# Install STA (Sliding Tile Attention)
RUN source $HOME/.local/bin/env && \
source /opt/venv/bin/activate && \
cd csrc/sliding_tile_attention/ && \
cd csrc/attn/ && \
git submodule update --init --recursive && \
python setup_sta.py install
# Install VSA
RUN source $HOME/.local/bin/env && \
source /opt/venv/bin/activate && \
cd csrc/sliding_tile_attention/ && \
cd csrc/attn/ && \
git submodule update --init --recursive && \
python setup_vsa.py install