torch
torchvision
xformers
accelerate
transformers
gradio
numpy