torch==2.0.1 
torchvision==0.15.2
transformers==4.29.2
diffusers[torch]==0.14.0
pynvml==11.4.1
bitsandbytes==0.38.1
ftfy==6.1.1
aiohttp==3.8.4
tensorboard>=2.11.0
protobuf==3.20.1
pyre-extensions==0.0.29
xformers==0.0.20
pytorch-lightning==1.6.5
OmegaConf==2.2.3
numpy==1.23.5
lion-pytorch
compel~=1.1.3
OmegaConf==2.2.3
numpy==1.23.5
wandb