torch
einops
flash_attn
Pillow
transformers
gradio