torch
transformers
torchvision
einops
safetensors
gradio==3.40.0