# GPU requirements for the slugline-mcp full reference index build.
#
# Targets a standard RunPod "PyTorch" template (CUDA 12.1 base image).
# If your pod's template already ships a working CUDA-enabled torch
# (check with `python3 -c "import torch; print(torch.__version__, torch.version.cuda)"`
# before installing anything), skip the torch line below and just install
# the rest -- reinstalling torch on top of a template's preinstalled build
# is the most common way to accidentally break GPU detection.
#
# If you do need to install torch (bare/minimal template, or CUDA version
# mismatch), uncomment the --extra-index-url line matching your pod's CUDA
# version (check with `nvidia-smi` -- top-right corner shows CUDA version).

# --extra-index-url https://download.pytorch.org/whl/cu121
# torch==2.4.1

transformers==4.44.2
accelerate==0.33.0
sentence-transformers==3.0.1
chromadb==0.5.5
datasets==2.20.0
huggingface-hub==0.24.6
