# NVIDIA Cosmos Predict2.5 & Transfer2.5 Dependencies
# Aligned with official requirements for ZeroGPU (H200 70GB)

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

# Diffusers with Cosmos support (>=0.34.0)
diffusers>=0.34.0

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

# HuggingFace utilities
huggingface-hub>=0.21.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 for UI
gradio>=5.0.0

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