[feat] add sliding_tile attention triton kernel and ROCM support (#916)

2026-01-22 12:06:38 +00:00
parent b824f9536b
commit 2f49e50150
7 changed files with 406 additions and 24 deletions
@@ -9,6 +9,7 @@ fastvideo generate \
    --model-path $MODEL_BASE \
    --sp-size ${num_gpus} \
    --tp-size 1 \
+    --num-gpus ${num_gpus} \
    --height 768 \
    --width 1280 \
    --num-frames 117 \