transformers
datasets
accelerate
gradio
torch
safetensors