fastapi
numpy
PyYAML
pyzmq
msgpack
python-multipart
requests
torch<2.13.0,>=2.9.1
torchvision>=0.24.1
torchaudio>=2.9.1
uvicorn
gdown
datasets

[all]
ruff
pytest
soundfile
av
diffusers>=0.39
fa3-fwd
einops
flashinfer-python>=0.6.4
huggingface-hub
mooncake-transfer-engine
Pillow
regex
safetensors
torchvision
torchaudio
torchcodec
transformers
triton
qwen-omni-utils
qwen-tts==0.1.1
numba>=0.61
ninja
accelerate
jiwer

[asr]
einops
flashinfer-python>=0.6.4
huggingface-hub
mooncake-transfer-engine
safetensors
torchcodec==0.8.*
transformers
datasets
jiwer

[audio]
soundfile

[bagel]
einops
flashinfer-python>=0.6.4
huggingface-hub
mooncake-transfer-engine
Pillow
regex
safetensors
torchvision
torchaudio
torchcodec
transformers

[bagel_xpu]
einops
huggingface-hub
mooncake-transfer-engine
Pillow
regex
safetensors
torchvision
torchaudio
torchcodec
transformers
vllm-xpu-kernels==0.1.13.2

[cosmos3]
av
diffusers>=0.39
fa3-fwd
flashinfer-python>=0.6.4
huggingface-hub
mooncake-transfer-engine
Pillow
safetensors
torchvision
torchcodec
transformers

[dev]
ruff<=0.16
pytest

[dynamo]
ai-dynamo<1.5,>=1.4.0
uvloop

[orpheus]
einops
flashinfer-python>=0.6.4
huggingface-hub
mooncake-transfer-engine
safetensors
transformers

[pi05]
flashinfer-python>=0.6.4
huggingface-hub
mooncake-transfer-engine
safetensors
transformers
triton

[qwen3_omni]
einops
flashinfer-python>=0.6.4
huggingface-hub
mooncake-transfer-engine
Pillow
regex
qwen-omni-utils
numba>=0.61
torchvision
torchaudio
torchcodec
transformers
datasets
ninja

[qwen3_tts]
einops
flashinfer-python>=0.6.4
huggingface-hub
mooncake-transfer-engine
qwen-tts==0.1.1
safetensors
transformers>=4.57.3

[vjepa2]
huggingface-hub
mooncake-transfer-engine
safetensors
torchcodec

[vjepa2_ac]
flashinfer-python>=0.6.4
huggingface-hub
mooncake-transfer-engine
safetensors
torchcodec

[wan22]
accelerate
diffusers>=0.39
huggingface-hub
transformers
av
