Spaces:

Cicici1109
/

IEAP

Running on Zero

Cicici1109 commited on 22 days ago

Commit

59ce571

verified ·

1 Parent(s): 26190ce

Update requirements.txt

Files changed (1) hide show

requirements.txt CHANGED Viewed

@@ -1,5 +1,5 @@
-torch==2.5.0 -f https://download.pytorch.org/whl/cu124/torch_stable.html
-torchvision==0.20.0 -f https://download.pytorch.org/whl/cu124/torch_stable.html
 diffusers==0.32.0
 transformers==4.42.3
 xtuner[deepspeed]==0.1.23
@@ -14,5 +14,5 @@ sentencepiece==0.2.0
 tornado==6.4.2
 openai==0.28.0
 gradio==5.32.0
-https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1%2Bcu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
 opencv-python

+torch==2.3.1 -f https://download.pytorch.org/whl/cu124/torch_stable.html
+torchvision==0.18.1 -f https://download.pytorch.org/whl/cu124/torch_stable.html
 diffusers==0.32.0
 transformers==4.42.3
 xtuner[deepspeed]==0.1.23
 tornado==6.4.2
 openai==0.28.0
 gradio==5.32.0
+https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1%2Bcu12torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
 opencv-python