transformers
torch
accelerate
flash-attn