Julian Bilcke
commited on
Commit
·
01fe26e
1
Parent(s):
829df0b
modify
Browse files- 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 |
|