requirements.txt

# Requirements for Hugging Face Spaces
# PyTorch with CUDA support - HF Spaces automatically provides CUDA-enabled PyTorch
# When you select GPU hardware, PyTorch will have full CUDA support
torch>=2.0.0
torchvision>=0.15.0
# Note: CUDA toolkit is pre-installed on HF Spaces GPU instances
# You can verify with: torch.cuda.is_available() and torch.cuda.get_device_name(0)

# Core dependencies
transformers>=4.30.0
accelerate>=0.20.0
huggingface_hub>=0.20.0
torchinfo
einops
scipy
sentencepiece
wandb[media]
torchmetrics[image]
simple_parsing
opencv-python
psutil
pyyaml
av==12.3.0

# Gradio for web interface
gradio>=6.0.2

# Git dependencies
git+https://github.com/KeKsBoTer/torch-dwt
git+https://github.com/huggingface/diffusers.git

# Video processing dependencies
decord
webdataset