[project]
name = "maxact-fast"
version = "0.1.0"
description = "General residual-direction -> text inverter: cluster-probe pretraining + Dr. GRPO RL"
requires-python = ">=3.11"
dependencies = [
    "torch>=2.8",
    "transformers>=4.57,<5",   # vllm-lens pin; keep <5
    "accelerate",              # required by device_map={"": device} in from_pretrained
    "peft",
    "datasets",
    "numpy",
    "scikit-learn",
    "wandb",
    "tqdm",
]

[project.optional-dependencies]
# installed in the dedicated vllm-lens venv, NOT the main env (hard pins)
vllm = ["vllm==0.19.0", "vllm-lens==1.1.0"]
cluster = ["faiss-gpu-cu12"]   # PyPI "faiss-gpu" is dead (py<=3.10 wheels only, conflicts with requires-python>=3.11)

[tool.setuptools.packages.find]
where = ["src"]