###############################################################################
#  CUDA 12.1 runtime  +  PyTorch 2.1 wheels  (official extra index)
###############################################################################
--extra-index-url https://download.pytorch.org/whl/cu121

torch==2.1.0+cu121            # core
torchvision==0.16.0+cu121     # must share the same ABI tag

###############################################################################
# Transformers / training utils
###############################################################################
transformers==4.46.3
accelerate>=0.26.0
bitsandbytes
einops
numpy<2.0.0

###############################################################################
# Video / image processing
###############################################################################
decord
imageio
imageio-ffmpeg
opencv-python-headless
ffmpeg-python
pillow

###############################################################################
# Gradio & web interface
###############################################################################
gradio==5.34.0
spaces
requests

###############################################################################
# HF ecosystem & model I/O
###############################################################################
huggingface_hub
sentencepiece
safetensors

###############################################################################
# Flash-Attention 2.5.8  ── prebuilt for
#   • CUDA 12.2 (binary still works under CUDA ≥12.1 drivers*)
#   • PyTorch 2.1
#   • Python 3.10  (cp310)
###############################################################################
https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.8/flash_attn-2.5.8+cu122torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl