pyproject.toml

[project]
name = "tame"
version = "0.1.0"
description = "Official implementation of LCMP & TAME (KDD 2026)"
readme = "README.md"
requires-python = "==3.13.*"
dependencies = [
    "colorama==0.4.6",
    "einops==0.8.1",
    "icecream==2.1.8",
    "loguru==0.7.3",
    "numpy==2.3.5",
    "openai==2.9.0",
    "pillow==12.0.0",
    "pydantic==2.12.5",
    "python-dotenv==1.2.1",
    "pyyaml==6.0.3",
    "qwen-vl-utils==0.0.14",
    "requests==2.32.5",
    "sentence-transformers==5.1.1",
    "torch==2.8.0",
    "torchvision==0.23.0",
    "tqdm==4.67.1",
    "transformers==4.57.0",
    "accelerate>=1.0",
    "flash-attn==2.8.3",
    "peft>=0.19.1",
    "timm>=1.0.27",
]

# This repo is a collection of scripts, not an installable package.
[tool.uv]
package = false

# flash-attn: use a prebuilt wheel (matches torch 2.8 + CUDA 12.8 + cp313 + linux x86_64)
# instead of compiling from source. From mjun0812/flash-attention-prebuild-wheels.
[tool.uv.sources]
flash-attn = { url = "https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.7.16/flash_attn-2.8.3%2Bcu128torch2.8-cp313-cp313-linux_x86_64.whl" }