importlib_metadata
filelock
httpx<1.0.0
huggingface-hub<2.0,>=0.34.0
numpy
regex!=2019.12.17
requests
safetensors>=0.8.0-rc.0
Pillow

[bitsandbytes]
bitsandbytes>=0.43.3
accelerate>=0.31.0

[dev]
urllib3<=2.0.0
isort>=5.5.4
ruff==0.9.10
hf-doc-builder>=0.3.0
compel==0.1.8
ftfy
GitPython<3.1.19
datasets
Jinja2
invisible-watermark>=0.2.0
librosa
parameterized
protobuf<4,>=3.20.3
pytest
pytest-timeout
pytest-xdist
requests-mock==1.10.0
safetensors>=0.8.0-rc.0
sentencepiece!=0.1.92,>=0.1.91
scipy
tiktoken>=0.7.0
torchsde
torchvision
transformers>=4.41.2
phonemizer
accelerate>=0.31.0
tensorboard
peft>=0.17.0
timm
torch>=1.4
jax>=0.4.1
jaxlib>=0.4.1
flax>=0.4.1

[docs]
hf-doc-builder>=0.3.0

[flashpack]
flashpack

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

[gguf]
gguf>=0.10.0
accelerate>=0.31.0

[nvidia_modelopt]
nvidia_modelopt[hf]>=0.33.1

[optimum_quanto]
optimum_quanto>=0.2.6
accelerate>=0.31.0

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

[test]
compel==0.1.8
ftfy
GitPython<3.1.19
datasets
Jinja2
invisible-watermark>=0.2.0
librosa
parameterized
protobuf<4,>=3.20.3
pytest
pytest-timeout
pytest-xdist
requests-mock==1.10.0
safetensors>=0.8.0-rc.0
sentencepiece!=0.1.92,>=0.1.91
scipy
tiktoken>=0.7.0
torchsde
torchvision
transformers>=4.41.2
phonemizer

[torch]
torch>=1.4
accelerate>=0.31.0

[torchao]
torchao>=0.7.0
accelerate>=0.31.0

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