docker/requirements.txt

setuptools>=61.0
torch==2.0.1
torchvision==0.15.2
transformers==4.32.0
albumentations==1.4.0
av==11.0.0
decord==0.6.0
einops==0.3.0
fastapi==0.110.0
accelerate==0.21.0
gdown==5.1.0
h5py==3.10.0
idna==3.6
imageio==2.34.0
matplotlib==3.7.5
numpy==1.24.4
omegaconf==2.1.1
opencv-python==4.9.0.80
opencv-python-headless==4.9.0.80
pandas==2.0.3
pillow==10.2.0
pydub==0.25.1
pytorch-lightning==1.4.2
pytorchvideo==0.1.5
PyYAML==6.0.1
regex==2023.12.25
requests==2.31.0
scikit-learn==1.3.2
scipy==1.10.1
six==1.16.0
tensorboard==2.14.0
test-tube==0.7.5
timm==0.9.16
torchdiffeq==0.2.3
torchmetrics==0.5.0
tqdm==4.66.2
urllib3==2.2.1
uvicorn==0.27.1
diffusers==0.24.0
scikit-video==1.1.11