transformers
datasets
gradio
torch
accelerate>=0.26.0