[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "smolomni"
version = "0.1.0"
description = "Tinman-SmolOmni-MLA: Unified any-to-any multimodal model with MLA attention"
readme = "README.md"
license = {text = "Apache-2.0"}
requires-python = ">=3.9"
authors = [
    {name = "Tinman Lab", email = "tinmanlab@example.com"}
]
keywords = [
    "multimodal", "vision-language-model", "mla-attention",
    "flow-matching", "image-generation", "on-device", "smol-scale"
]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Developers",
    "Intended Audience :: Science/Research",
    "License :: OSI Approved :: Apache Software License",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.9",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]

dependencies = [
    "torch>=2.0.0",
    "transformers>=4.40.0",
    "accelerate>=0.25.0",
    "huggingface-hub>=0.20.0",
    "numpy>=1.24.0",
    "safetensors>=0.4.0",
    "tqdm",
]

[project.optional-dependencies]
audio = ["soundfile", "librosa", "torchaudio"]
onnx = ["onnx", "onnxruntime"]
quantization = ["bitsandbytes"]
dev = ["pytest", "black", "ruff"]
all = [
    "smolomni[audio]",
    "smolomni[onnx]",
    "smolomni[quantization]",
]

[project.urls]
Homepage = "https://huggingface.co/TinmanLabSL/SmolOmni-MLA-Toolkit"
Repository = "https://huggingface.co/TinmanLabSL/SmolOmni-MLA-Toolkit"
Documentation = "https://huggingface.co/TinmanLabSL/SmolOmni-MLA-Toolkit/blob/main/README.md"

[tool.setuptools.packages.find]
where = ["."]
include = ["smolomni*"]

[tool.black]
line-length = 100
target-version = ['py39', 'py310', 'py311', 'py312']

[tool.ruff]
line-length = 100
select = ["E", "F", "W", "I"]
ignore = ["E501"]