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

[dev]
build
twine

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