transformers
torch
flash-attn