Spaces:

TencentARC
/

RollingForcing

Running on Zero

kunhaokhliu commited on Oct 31

Commit

a85f330

1 Parent(s): 4e71303

update

Files changed (1) hide show

requirements.txt CHANGED Viewed

@@ -41,4 +41,4 @@ torchao
 tensorboard
 ninja
 packaging
-https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.0.0/flash_attn-2.6.3+cu124torch2.5-cp312-cp312-linux_x86_64.whl

 tensorboard
 ninja
 packaging
+https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu118torch1.12cxx11abiFALSE-cp310-cp310-linux_x86_64.whl