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

[VSA] [STA] Fix directory structure for pypi publishing (#769)

Browse Source 
Co-authored-by: Peiyuan Zhang <a1286225768@gmail.com>
This commit is contained in:
William Lin
2025-08-28 22:34:03 -07:00
committed by GitHub
co-authored by Peiyuan Zhang
parent 2dcc5ea4f6
commit b91020b407
30 changed files with 90 additions and 98 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile.python3.11
+5 -5
View File
@@ -58,15 +58,15 @@ RUN source $HOME/.local/bin/env && \
# Install STA (Sliding Tile Attention)
RUN source $HOME/.local/bin/env && \
source /opt/venv/bin/activate && \
cd csrc/attn && \
cd csrc/attn/sliding_tile_attn && \
git submodule update --init --recursive && \
python setup_sta.py install
python setup.py install
# Install VSA
RUN source $HOME/.local/bin/env && \
source /opt/venv/bin/activate && \
cd csrc/attn && \
cd csrc/attn/video_sparse_attn && \
git submodule update --init --recursive && \
python setup_vsa.py install
python setup.py install
EXPOSE 22
EXPOSE 22