importlib_metadata
filelock
huggingface-hub>=0.23.2
numpy
regex!=2019.12.17
requests
safetensors>=0.3.1
Pillow

[dev]
urllib3<=2.0.0
isort>=5.5.4
ruff==0.1.5
hf-doc-builder>=0.3.0
compel==0.1.8
GitPython<3.1.19
datasets
Jinja2
invisible-watermark>=0.2.0
k-diffusion>=0.0.12
librosa
parameterized
pytest
pytest-timeout
pytest-xdist
requests-mock==1.10.0
safetensors>=0.3.1
sentencepiece!=0.1.92,>=0.1.91
scipy
torchvision
transformers>=4.41.2
accelerate>=0.31.0
protobuf<4,>=3.20.3
tensorboard
peft>=0.6.0
torch>=1.4
jax>=0.4.1
jaxlib>=0.4.1
flax>=0.4.1

[docs]
hf-doc-builder>=0.3.0

[flax]
jax>=0.4.1
jaxlib>=0.4.1
flax>=0.4.1

[quality]
urllib3<=2.0.0
isort>=5.5.4
ruff==0.1.5
hf-doc-builder>=0.3.0

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

[torch]
torch>=1.4
accelerate>=0.31.0

[training]
accelerate>=0.31.0
datasets
protobuf<4,>=3.20.3
tensorboard
Jinja2
peft>=0.6.0