transformers>=5.17.0
accelerate
torchvision
pillow
flash-linear-attention
torchao