numpy
Pillow
Requests
flash_attn
accelerate>=0.25.0
huggingface-hub
transformers>=4.44.0
einops>=0.7.0
torch>=2.1.2
torchvision>=0.16.2
pyvips>=2.1.16