setuptools>=67.0
psutil
wheel
click
termcolor
tabulate
lz4
six
tqdm
rich
diffusers>=0.17.1
transformers
datasets
evaluate
huggingface_hub

[dev]
build
twine

[test]
pytest
pytest-cov
mypy
isort
black
ruff
gpustat
