torch==2.1.0 
torchvision==0.16.0
transformers>=4.38.2
diffusers[torch]==0.21.4
peft>=0.9.0
pynvml==11.4.1
bitsandbytes==0.41.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.22.post7
pytorch-lightning==1.6.5
OmegaConf==2.2.3
numpy==1.23.5
dowg
lion-pytorch
compel~=1.1.3
OmegaConf==2.2.3
numpy==1.23.5
wandb
colorama
safetensors
torchsde