torch==2.0.1
torchvision==0.15.2
xformers
bitsandbytes
transformers
accelerate
diffusers==0.23.1
scipy
huggingface_hub
datasets