torch
transformers
accelerate
flash-attn --no-build-isolation