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