torch
xformers
torchvision
einops
Pillow
requests
datasets
transformers
huggingface_hub==0.19.4
hf-transfer==0.1.4