torch
transformers
gradio
safetensors
accelerate