torch
transformers
accelerate
flash-attn