# ── Core framework ─────────────────────────────────────────────────────────────
fastapi>=0.111.0
uvicorn[standard]>=0.29.0
pydantic>=2.7.0

# ── ML / model loading ─────────────────────────────────────────────────────────
transformers>=4.40.0
sentencepiece>=0.2.0        # required by T5Tokenizer
protobuf>=4.25.0            # required by sentencepiece / T5
accelerate>=0.30.0          # optional but strongly recommended for large models
safetensors>=0.4.3          # fast model loading from .safetensors files

# ── Numerics ────────────────────────────────────────────────────────────────────
numpy>=1.26.0

# ── PyTorch — choose ONE of the blocks below and uncomment it ──────────────────

# NVIDIA CUDA (CUDA 12.1)  ← default for most NVIDIA GPUs
torch>=2.3.0
torchvision>=0.18.0
# Install with:
#   pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121

# AMD ROCm 6.x  ← uncomment and use for AMD GPUs
# torch>=2.3.0
# torchvision>=0.18.0
# Install with:
#   pip install torch torchvision --index-url https://download.pytorch.org/whl/rocm6.0

# CPU-only fallback  ← uncomment for CPU-only
# torch>=2.3.0
# torchvision>=0.18.0
# Install with:
#   pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
