setuptools>=61.0
accelerate
diffusers
matplotlib
pandas
pyarrow
timm
torch
torchmetrics
torchvision
transformers
