requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu130

# Web UI
gradio==5.50.0

# Core numeric / vision
numpy
opencv-python

# PyTorch
torch
torchvision

# Hugging Face helper for downloading weights
huggingface-hub==0.36.0

# Utilities
tqdm

# SeedVR2
psutil
einops
diffusers
rotary-embedding-torch
omegaconf

gguf

triton; sys_platform != 'win32'
triton-windows; sys_platform == 'win32'

# 
# flash-attn; sys_platform != 'win32'
# sageattention; sys_platform != 'win32'
# 
# if sys_platform == 'win32'
# https://huggingface.co/lldacing/flash-attention-windows-wheel
# https://huggingface.co/ussoewwin/Flash-Attention-2_for_Windows
# https://github.com/woct0rdho/SageAttention