Pillow
filelock
huggingface-hub>=0.13.2
importlib_metadata
numpy
regex!=2019.12.17
requests

[dev]
Jinja2
Jinja2
accelerate>=0.11.0
accelerate>=0.11.0
black~=23.1
compel==0.1.8
datasets
datasets
flax>=0.4.1
hf-doc-builder>=0.3.0
hf-doc-builder>=0.3.0
invisible-watermark
isort>=5.5.4
jax!=0.3.2,>=0.2.8
jaxlib>=0.1.65
k-diffusion>=0.0.12
librosa
omegaconf
parameterized
protobuf<4,>=3.20.3
pytest
pytest-timeout
pytest-xdist
requests-mock==1.10.0
ruff>=0.0.241
safetensors
scipy
sentencepiece!=0.1.92,>=0.1.91
tensorboard
torch>=1.4
torchvision
transformers>=4.25.1
urllib3<=2.0.0

[docs]
hf-doc-builder>=0.3.0

[flax]
flax>=0.4.1
jax!=0.3.2,>=0.2.8
jaxlib>=0.1.65

[quality]
black~=23.1
hf-doc-builder>=0.3.0
isort>=5.5.4
ruff>=0.0.241
urllib3<=2.0.0

[test]
Jinja2
compel==0.1.8
datasets
invisible-watermark
k-diffusion>=0.0.12
librosa
omegaconf
parameterized
pytest
pytest-timeout
pytest-xdist
requests-mock==1.10.0
safetensors
scipy
sentencepiece!=0.1.92,>=0.1.91
torchvision
transformers>=4.25.1

[torch]
accelerate>=0.11.0
torch>=1.4

[training]
Jinja2
accelerate>=0.11.0
datasets
protobuf<4,>=3.20.3
tensorboard
