torch==2.1.0
torchvision
torchaudio
transformers
diffusers
gradio
xformers
accelerate
numpy<2