torch
transformers
torchvision
einops
safetensors
gradio==4.40.0