torch
diffusers
transformers
accelerate
einops
kornia
gradio
triton
torchvision
opencv-python
xformers