torch>=2.3.1
torchvision>=0.10.0
pytorch-lightning>=1.8.6
PyYAML>=6.0
lightning>=2.2.5
lightning-lite>=1.8.6
lightning-utilities>=0.8.0
hydra-colorlog>=1.2.0
hydra-core>=1.1.1
deepspeed>=0.15.0
tensorboard>=2.13.0
tensorboardX>=2.6.2
tqdm>=4.65.0
transformers @ git+https://github.com/huggingface/transformers.git@84f0186e8971a21bcda9b446a8a74f0f1a958f1c
diffusers>=0.31.0
wandb>=0.19.0
numpy<2.0
sentence-transformers==2.2.2
open_clip_torch==2.20.0
datasets==2.12.0
dlimp @ git+https://github.com/kvablack/dlimp@5edaa4691567873d495633f2708982b42edf1972
ml_collections==1.0.0
accelerate>=0.26.0
tensorflow-graphics==2021.12.3
imageio[ffmpeg]
robosuite==1.4.1
bddl
easydict
cloudpickle
gym
num2words
draccus
peft