use fa3 in sgl-kernel (#4954)

This commit is contained in:
Yineng Zhang
2025-03-31 16:14:49 -07:00
committed by GitHub
parent ee47a6c1c3
commit 1c63e79756
3 changed files with 3 additions and 3 deletions

View File

@@ -15,7 +15,7 @@ pip install -e "python[all]" --find-links https://flashinfer.ai/whl/cu124/torch2
rm -rf /root/.cache/flashinfer
# Force reinstall flashinfer and torch_memory_saver
pip install flashinfer_python==0.2.3 --find-links ${FLASHINFER_REPO} --force-reinstall --no-deps
pip install sgl-kernel==0.0.5.post4 --force-reinstall
pip install sgl-kernel==0.0.6 --force-reinstall
pip install torch_memory_saver
pip install transformers==4.50.0 sentence_transformers accelerate==1.4.0 peft pandas datasets timm torchaudio