accelerate<2.0.0,>=1.10.0
av>=14.2.0
diffusers==0.35.1
draccus==0.10.0
einops>=0.8.0
huggingface-hub[cli]<1.0,>=0.34
msgpack
numpy==1.26.4
packaging>=24.2
pillow>=10
pyyaml
safetensors
simplejson>=3.19
torch==2.7.1
torchdata<1.0,>=0.8.0
torchvision==0.22.1
tqdm
transformers==4.57.6
websockets

[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.7.1
torchcodec==0.5
