transformers
torch
accelerate
#flash-attn