mirror of
https://github.com/hpcaitech/Open-Sora.git
synced 2026-04-10 12:49:38 +02:00
29 lines
287 B
Text
29 lines
287 B
Text
# should be installed by users
|
|
torch>=1.13
|
|
colossalai
|
|
flash_attn # optional
|
|
apex # optional
|
|
|
|
# packages
|
|
accelerate
|
|
diffusers
|
|
timm
|
|
transformers
|
|
xformers # necessary for PixArt
|
|
|
|
# config & logging
|
|
gdown
|
|
pre-commit
|
|
matplotlib
|
|
mmengine
|
|
tensorboard
|
|
tqdm
|
|
wandb
|
|
|
|
# video
|
|
pyav
|
|
|
|
# text
|
|
clip
|
|
ftfy
|