| [build-system] | |
| requires = ["setuptools>=69", "wheel"] | |
| build-backend = "setuptools.build_meta" | |
| [project] | |
| name = "logits-sigma-1-duo" | |
| version = "0.1.0" | |
| description = "Decision Index kit engine for Logits Sigma-1 Duo: two decision models on one GPU, option probabilities averaged." | |
| requires-python = ">=3.12" | |
| license = {text = "Apache-2.0"} | |
| # Exact versions: requirements.lock next to this file (install it first). The kit is installed separately, without | |
| # its dependencies: pip install --no-deps "decision-index @ git+https://github.com/apolinario/decision-index@25c6ec7". | |
| # d3-ultra's engine (d3_engine.py) is imported from the model repo's d3-ultra/ directory, not from this package. | |
| dependencies = [ | |
| "torch==2.15.0", | |
| "transformers==5.22.1", | |
| "flash-linear-attention==0.6.0", | |
| "safetensors>=0.8", | |
| "numpy>=2", | |
| ] | |
| [project.optional-dependencies] | |
| # compiled from source (nvcc 13 + g++); the Dockerfile builds it for sm_90 and sm_120 | |
| conv = ["causal-conv1d==1.7.0"] | |
| [tool.setuptools.packages.find] | |
| include = ["logits_sigma_1_duo*"] | |