torch
gradio
transformers
accelerate==0.26.0