huggingface_hub>=0.21.0
numpy<3.0.0,>=1.17
packaging>=20.0
psutil
pyyaml
safetensors>=0.4.3
torch>=1.10.0

[deepspeed]
deepspeed

[dev]
bitsandbytes
black~=23.1
datasets
diffusers
evaluate
hf-doc-builder>=0.3.0
parameterized
pytest-subtests
pytest-xdist
pytest<=8.0.0,>=7.2.0
rich
ruff~=0.6.4
scikit-learn
scipy
timm
torchdata>=0.8.0
torchpippy>=0.2.0
tqdm
transformers

[docs]

[quality]
black~=23.1
hf-doc-builder>=0.3.0
ruff~=0.6.4

[rich]
rich

[sagemaker]
sagemaker

[test_dev]
bitsandbytes
datasets
diffusers
evaluate
scikit-learn
scipy
timm
torchdata>=0.8.0
torchpippy>=0.2.0
tqdm
transformers

[test_prod]
parameterized
pytest-subtests
pytest-xdist
pytest<=8.0.0,>=7.2.0

[test_trackers]
comet-ml
dvclive
tensorboard
wandb

[testing]
bitsandbytes
datasets
diffusers
evaluate
parameterized
pytest-subtests
pytest-xdist
pytest<=8.0.0,>=7.2.0
scikit-learn
scipy
timm
torchdata>=0.8.0
torchpippy>=0.2.0
tqdm
transformers
