requirements.txt

# NeuroScope-Web Backend
# TransformerLens-powered interpretability engine

# Core ML
torch==2.3.0
transformer-lens==2.11.0

# transformer-lens 2.11 imports `transformers.TRANSFORMERS_CACHE` at module
# load; that attribute was removed in transformers 5.x. Bound the major
# version. huggingface-hub<1 keeps the legacy 0.x API surface TL expects.
transformers>=4.37.2,<5.0
huggingface-hub<1.0

# API
fastapi==0.115.0
uvicorn[standard]==0.30.6
pydantic==2.9.2

# Analysis
numpy==1.26.4
scikit-learn==1.5.2

# Used by scripts/build_refusal_pairs.py (one-time local script, not at runtime)
# Not strictly needed by the API server, but pinned here so the script
# works from a fresh `pip install -r requirements.txt`.
datasets>=2.18,<5.0