accelerate>=1.9
diffusers>=0.36
einops
huggingface-hub>=0.34
peft>=0.17
protobuf
sentencepiece
torchvision>=0.20
transformers>=4.54

[ci]
accelerate==1.9
cache-dit==1.1.9
controlnet-aux==0.0.10
datasets==3.6
diffusers==0.36
huggingface-hub==0.34
peft==0.17
pytest==8.4.2
pytest-rerunfailures==16.0.1
torchmetrics==1.8
transformers==4.55.2

[demo]
controlnet-aux
gradio==5.39
spaces

[dev]
cache-dit>=1.1.9
controlnet-aux
datasets<4
pre-commit
pytest
pytest-rerunfailures
torchmetrics

[docs]
docutils<0.21
furo
graphviz
ipykernel
jupyter
myst-parser
nbsphinx
sphinx
sphinx-book-theme
sphinx-copybutton
sphinx-rtd-theme
sphinx-tabs
sphinxawesome-theme
sphinxcontrib-mermaid
sphinxext-rediraffe

[full]
controlnet-aux
