# GPU-accelerated PyTorch dependencies for Rex AI Assistant (CUDA 11.8)
# Use this for CUDA 11.8 environments (Linux). For CUDA 12.4, use requirements-gpu-cu124.txt.

# PyTorch CUDA 11.8 versions (compatible set)
# Keep the torch/torchvision/torchaudio trio on the same verified 2.6.x line.
--extra-index-url https://download.pytorch.org/whl/cu118
torch==2.12.0
torchvision==0.21.0+cu118
torchaudio==2.6.0+cu118

# transformers: pin to a semver-compatible range for Coqui TTS 0.22.0.
# BeamSearchScorer was removed in 4.38+; the rex/compat/ shim restores it.
transformers>=4.38.0,<5.0

# Install Rex with ML + audio extras
-e ".[ml,audio]"