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