transformers
scipy
accelerate
torchaudio
torch
torchvision
flash-attn
packaging
ninja