File size: 763 Bytes
8505f8e 48db85f 8505f8e 48db85f 8505f8e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | [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"]
|