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

update docker

Browse Source
This commit is contained in:
SolitaryThinker
2025-12-24 09:32:42 +00:00
parent f8bfc76015
commit 104a539a22
4 changed files with 9 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile.python3.11
+2 -9
View File
@@ -55,17 +55,10 @@ RUN source $HOME/.local/bin/env && \
echo 'source /opt/venv/bin/activate' >> /root/.bashrc && \
echo 'if [ -n "$ZSH_VERSION" ] && [ -f ~/.zshrc ]; then . ~/.zshrc; elif [ -f ~/.bashrc ]; then . ~/.bashrc; fi' > /root/.profile
# Install STA (Sliding Tile Attention)
# Install FastVideo Kernels
RUN source $HOME/.local/bin/env && \
source /opt/venv/bin/activate && \
cd csrc/attn/sliding_tile_attn && \
git submodule update --init --recursive && \
python setup.py install
# Install VSA
RUN source $HOME/.local/bin/env && \
source /opt/venv/bin/activate && \
cd csrc/attn/video_sparse_attn && \
cd csrc/fastvideo_kernel && \
git submodule update --init --recursive && \
python setup.py install