setuptools>=61.0
accelerate
bitsandbytes
black
diffusers
ftfy
isort
lightning
matplotlib
numpy
pandas
PyYAML
pillow
regex
requests
ruff
sentencepiece
tensorboard
timm
torch
torchmetrics
torchvision
tqdm
transformers
