diffusers>=0.36.0
transformers>=5.6.0
hf_transfer>=0.1.0
datasets>=3.0.1
wandb>=0.21.0
requests>=2.32.4
pillow>=11.3.0
trainingsample>=0.2.10
accelerate>=1.5.2
safetensors>=0.5.3
open-clip-torch>=2.26.1
iterutils>=0.1.6
scipy>=1.11.1
boto3>=1.35.83
pandas>=2.2.3
botocore>=1.35.83
skrample<0.6.0,>=0.5.3
urllib3<1.27
torchsde>=0.2.6
torchmetrics>=1.1.1
colorama>=0.4.6
numpy>=2.2.0
num2words>=0.5.13
peft>=0.18.0
tensorboard>=2.18.0
py3langid>=0.2.2
pypinyin>=0.50.0
sentencepiece>=0.2.0
spacy>=3.7.4
hangul-romanize>=0.1.0
optimum-quanto>=0.2.7
lycoris-lora>=3.4.0
torch-optimi>=0.2.1
librosa>=0.10.2
loguru>=0.7.2
toml>=0.10.2
fastapi[standard]>=0.115.0
sse-starlette>=1.6.5
atomicwrites>=1.4.1
beautifulsoup4>=4.12.3
prodigy-plus-schedule-free>=1.9.2
tokenizers>=0.21.0
huggingface-hub>=0.34.3
imageio-ffmpeg>=0.6.0
imageio[pyav]>=2.37.0
hf-xet>=1.1.5
peft-singlora>=0.2.0
vector-quantize-pytorch>=1.27.15
cryptography>=41.0.0
torchcodec>=0.10.0
sdnq>=0.1.2
aiosqlite>=0.19.0
httpx>=0.28.0
psutil>=5.9.0
webshart>=0.4.6

[all]
pillow-jxl-plugin>=1.3.1
selenium>=4.0.0
coverage>=7.0.0
black>=23.0.0
isort>=5.12.0
flake8>=6.0.0

[apple]
torch>=2.10.0
torchvision>=0.25.0
torchaudio>=2.10.0
torchao<0.16.0,>=0.14.0

[captioning]
caption-flow[vllm]>=0.5.2

[captioning-cuda13]
caption-flow>=0.5.2
numpy<2.3.0,>=2.2.0
transformers<6.0.0,>=5.5.1
vllm<0.20.0,>=0.19.1

[cpu]
torch>=2.10.0
torchvision>=0.25.0
torchaudio>=2.10.0
torchao<0.16.0,>=0.14.0

[cuda]
torch>=2.10.0
torchvision>=0.25.0
torchaudio>=2.10.0
triton>=3.3.0
bitsandbytes>=0.45.0
deepspeed>=0.17.2
torchao<0.16.0,>=0.14.0
nvidia-cudnn-cu12
nvidia-nccl-cu12
nvidia-ml-py>=12.555
lm-eval>=0.4.4
ramtorch

[cuda13]
torch>=2.10.0
torchvision>=0.25.0
torchaudio>=2.10.0
triton>=3.3.0
deepspeed>=0.17.2
torchao<0.16.0,>=0.14.0
bitsandbytes>=0.45.0
nvidia-cudnn-cu13
nvidia-nccl-cu13
nvidia-ml-py>=12.555
lm-eval>=0.4.4
ramtorch

[cuda13-captioning]
caption-flow>=0.5.2
numpy<2.3.0,>=2.2.0
transformers<6.0.0,>=5.5.1
vllm<0.20.0,>=0.19.1

[dev]
selenium>=4.0.0
coverage>=7.0.0
black>=23.0.0
isort>=5.12.0
flake8>=6.0.0
mypy>=1.0.0
pre-commit>=3.0.0

[docs]
zensical>=0.0.19

[jxl]
pillow-jxl-plugin>=1.3.1

[rocm]
torch>=2.10.0
torchvision>=0.25.0
torchaudio>=2.10.0
triton>=3.3.0
torchao<0.16.0,>=0.14.0
ramtorch

[state-all]
asyncpg>=0.29.0
aiomysql>=0.2.0
redis>=5.0.0

[state-mysql]
aiomysql>=0.2.0

[state-postgresql]
asyncpg>=0.29.0

[state-redis]
redis>=5.0.0

[test]
selenium>=4.0.0
coverage>=7.0.0
