gradio
transformers
torch
accelerate
flash-attention