| [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"] | |