torch==2.2.1
torchvision==0.17.1
tensorflow==2.15.0
tensorflow-datasets==4.9.3
transformers==4.38.2
diffusers==0.27.0
imageio==2.9.0
imageio-ffmpeg==0.4.4
piqa
lpips
peft==0.10.0
opencv-python
accelerate
wandb
nvitop
huggingface_hub==0.25.2 -i https://pypi.tuna.tsinghua.edu.cn/simple
matplotlib
xformers == 0.0.25
datasets >= 2.14.0
sentencepiece != 0.1.92
protobuf
evaluate
scikit-learn
hydra-core
dm_env
gym
hydra-submitit-launcher
PyOpenGL_accelerate
bitsandbytes