requirements.txt

# Orchestrator + MDM family — single venv on HF Space.
# Outlier models (Kimodo etc.) live in their own venvs (escape_hatch) and are
# subprocess-called from inside @spaces.GPU.

# Core orchestration.
# NOTE: gradio is intentionally NOT pinned here. HF's Gradio SDK runtime
# injects its own gradio version (currently 6.13.0); pinning a different
# version causes a build conflict. For OSS-local dev we accept whatever
# pip resolves alongside HF's other deps.
fastapi>=0.115.6
uvicorn[standard]>=0.32.1
# gradio — injected by HF Gradio SDK runtime
httpx>=0.27.0
python-multipart>=0.0.20
slowapi>=0.1.9
websocket-client>=1.8.0
pydantic>=2.10.4

# HF Spaces ZeroGPU runtime — no-op in OSS mode
# spaces — injected by HF Gradio SDK runtime

# huggingface_hub — used at Docker build time to download baked-in model
# checkpoints from AnimoFlow/animoflow-checkpoints. Also used at runtime
# by transformers.
huggingface_hub>=0.27.0

# Numerics for IK + retarget pipeline (matches start-local.sh's pinned trio)
# torch — injected by HF ZeroGPU runtime
# torchvision — injected by HF ZeroGPU runtime
numpy>=1.26.4
scipy>=1.13.1
matplotlib>=3.9.2

# MDM-family deps (CLIP, transformers used by MDM/priorMDM/MoMask)
transformers>=4.46.3
ftfy>=6.3.1
regex>=2024.11.6
git+https://github.com/openai/CLIP.git
# Pulled from comfyui-animoflow/containers/mdm/requirements.txt — adding here
# rather than installing MDM container's reqs separately, because that file
# pins torch==2.3.0 which downgrades us from 2.4.1 and removes the
# torch.library.register_fake API the upstream MDM source code uses.
einops>=0.8.0

# Blender retarget pipeline support
trimesh>=4.5.3

# Multilingual prompt rewriter (animoflow-api/api/rewriter.py)
sentence-transformers>=3.0
accelerate>=1.0

# Tests
pytest>=8.3.4
pytest-asyncio>=0.25.0