# Main dependencies
torch>=2.6.0
torchvision>=0.21.0
lightning>=2.2.4
numpy>=1.26.4
matplotlib>=3.8.0
pyyaml>=6.0.0

# Huggingface
huggingface-hub>=0.30.2
safetensors

# Image processing and computer vision
kornia>=0.7.2
opencv-python>=4.10.0.84
opencv-contrib-python>=4.10.0.84
einops>=0.8.0

# Custom cuda implementation /modules/cupy_module/
cupy-cuda12x>=12.0.0  # For CUDA 12.4
# Note: For cupy, you need to install the specific version for your CUDA version
# Examples:
# cupy-cuda11x for CUDA 11.x
# cupy-cuda12x for CUDA 12.x
# cupy-cuda10x for CUDA 10.x

# Utilities and tools
scipy>=1.7.0
tensorboard>=2.8.0

# Project-Specific Dependencies
# RAFT (Flow Estimation)
# Note: RAFT is included in the project code; no external installation is required.

# FLOLPIPS (Quality Metrics)
# Note: FLOLPIPS is included in the project code; no external installation is required.

# Gradio
gradio>=4.34.0
imageio>=2.34.1
imageio-ffmpeg>=0.6.0