transformers
torch
pillow
accelerate
flash-attn