jbilcke-hf HF Staff commited on
Commit
01fe26e
·
1 Parent(s): 829df0b
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -31,7 +31,7 @@ torchdata==0.10.1
31
  torchao==0.9.0
32
 
33
  # for torch 2.6, we must use torchcodec 0.2
34
- --index-url=https://download.pytorch.org/whl/cu128
35
  torchcodec==0.2.1
36
  flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
37
 
 
31
  torchao==0.9.0
32
 
33
  # for torch 2.6, we must use torchcodec 0.2
34
+ --extra-index-url=https://download.pytorch.org/whl/cu128
35
  torchcodec==0.2.1
36
  flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
37