# NVIDIA CUDA — install with:
# pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu128
# Then install base.txt
--index-url https://download.pytorch.org/whl/cu128
torch
torchaudio

# Optional: flash-attn for faster inference (requires CUDA toolkit / nvcc)
# Install CUDA toolkit first: sudo pacman -S cuda  (CachyOS/Arch)
# Then: CUDA_HOME=/opt/cuda pip install flash-attn --no-build-isolation
