Open-Sora/requirements.txt
Zheng Zangwei (Alex Zheng) febf3ad4b2
Update Open-Sora 2.0 (#807)
* upload v2.0

* update docs

* [hotfix] fit latest fa3 (#802)

* update readme

* update readme

* update readme

* update train readme

* update readme

* update readme: motion score

* cleaning video dc ae WIP

* update config

* add dependency functions

* undo cleaning

* use latest dcae

* complete high compression training

* update hcae config

* cleaned up vae

* update ae.md

* further cleanup

* update vae & ae paths

* align naming of ae

* [hotfix] fix ring attn bwd for fa3 (#803)

* train ae default without wandb

* update config

* update evaluation results

* added hcae report

* update readme

* update readme demo

* update readme demo

* update readme gif

* display demo directly in readme

* update paper

* delete files

---------

Co-authored-by: Hongxin Liu <lhx0217@gmail.com>
Co-authored-by: Shen-Chenhui <shen_chenhui@u.nus.edu>
Co-authored-by: wuxiwen <wuxiwen.simon@gmail.com>
2025-03-12 13:14:22 +08:00

17 lines
293 B
Text

torch==2.4.0
torchvision==0.19.0
colossalai>=0.4.4
mmengine>=0.10.3
ftfy>=6.2.0 # for t5
accelerate>=0.29.2 # for t5
av==13.1.0 # for video loading
liger-kernel==0.5.2
pandas>=2.0.3
pandarallel>=1.6.5
openai>=1.52.2
wandb>=0.17.0
tensorboard>=2.14.0
pre-commit>=3.5.0
omegaconf>=2.3.0
pyarrow