nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-21 11:59:01 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

update requirements

Browse source
This commit is contained in:
zhengzangw 2024-06-24 09:17:32 +00:00
parent f40ea2270c
commit 7b30ede7f0
2 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
requirements/requirements-cu121.txt
View file

@ -1,3 +1,3 @@
torch==2.3.0 --index-url https://download.pytorch.org/whl/cu121
torchvision==0.18.0 --index-url https://download.pytorch.org/whl/cu121
xformers==0.0.26.post1 --index-url https://download.pytorch.org/whl/cu121
torch==2.2.2 --index-url https://download.pytorch.org/whl/cu121
torchvision==0.17.2 --index-url https://download.pytorch.org/whl/cu121
xformers==0.0.25.post1 --index-url https://download.pytorch.org/whl/cu121

3
requirements/requirements.txt
View file

@ -1,4 +1,4 @@
colossalai==0.3.7
colossalai==0.3.9
mmengine>=0.10.3
pandas>=2.0.3
timm==0.9.16
@ -7,6 +7,7 @@ ftfy>=6.2.0 # for t5
diffusers==0.27.2 # for vae
accelerate==0.29.2 # for t5
av>=12.0.0 # for video loading
numpy<2.0.0
# [gradio]
gradio>=4.26.0