Update requirements.txt
Browse files- requirements.txt +0 -2
requirements.txt
CHANGED
@@ -2,6 +2,4 @@ numpy==1.25.0
|
|
2 |
torch==2.4.0
|
3 |
transformers==4.46.2
|
4 |
pillow==10.3.0
|
5 |
-
# https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.0.post2/flash_attn-2.7.0.post2+cu11torch2.4cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
|
6 |
-
# https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.0.post2/flash_attn-2.7.0.post2+cu11torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|
7 |
https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.3/flash_attn-2.6.3+cu123torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|
|
|
2 |
torch==2.4.0
|
3 |
transformers==4.46.2
|
4 |
pillow==10.3.0
|
|
|
|
|
5 |
https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.3/flash_attn-2.6.3+cu123torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|