accelerate==1.15.0
av>=14.2.0
diffusers==0.30.3
draccus==0.10.0
einops==0.9.0.dev0
huggingface-hub[cli]==0.36.2
msgpack==1.2.3
numpy==1.26.4
packaging==26.3
pillow==12.3.0
pyyaml==6.0.3
safetensors==0.9.0rc1
simplejson==4.2.0
tokenizers==0.21.4
torch==2.8.0
torchdata==0.11.0
torchvision==0.23.0
tqdm==4.70.1
transformers==4.51.3
websockets>=13.1

[:sys_platform == "linux" and platform_machine == "x86_64"]
nvidia-cublas-cu12==12.8.4.1
nvidia-cudnn-cu12==9.10.2.21

[train]
blobfile>=3.0.0
datasets==3.6.0
gymnasium==0.29.1
matplotlib
numpydantic
omegaconf>=2.3.0
tensorboard==2.16.2
termcolor>=2.4.0
torchaudio==2.8.0
torchcodec==0.6.0
