# NVIDIA Cosmos Predict2.5 & Transfer2.5 Dependencies
# Aligned with official requirements for ZeroGPU (H200 70GB)
# Build timestamp: 2026-01-27T12:00:00 - force rebuild v2

# Core PyTorch stack
torch==2.5.1
torchvision==0.20.1

# Diffusers with Cosmos 2.5 support (main branch for Cosmos2_5_PredictBasePipeline)
git+https://github.com/huggingface/diffusers.git

# Transformers ecosystem
transformers>=4.52.4
accelerate>=1.7.0
tokenizers>=0.21.0
safetensors>=0.4.0

# HuggingFace utilities (pin <1.0 to keep HfFolder for Gradio)
huggingface-hub>=0.34.0,<1.0.0
hf_transfer>=0.1.4

# Video processing
av>=14.0.0
opencv-python-headless>=4.8.0
imageio>=2.31.0
imageio-ffmpeg>=0.4.9

# Core dependencies
einops>=0.8.1
pillow>=10.0.0
numpy>=1.24.0
scipy>=1.10.0

# Metrics for validation
scikit-image>=0.21.0
lpips>=0.1.4

# Gradio version is controlled by README.md sdk_version (4.44.0)
# Do NOT specify gradio here - it will be overridden by sdk_version

# Note: cosmos_guardrail is provided as a local mock module
# to avoid version conflicts (official package requires torch==2.6.0)

# Note: spaces package is pre-installed in ZeroGPU environment