torch
torchvision
flash-attn
transformers
accelerate
bitsandbytes