docker/requirements.txt

accelerate==0.23.0
einops==0.7.0
diffusers==0.21.4
transformers==4.34.0
wandb==0.15.11
moviepy==1.0.3
imageio==2.31.5
xformers==0.0.22