[project]
name = "ltx-core"
version = "0.1.0"
description = "Core implementation of Lightricks' LTX-2 model"
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
    "torch~=2.7",
    "torchaudio",
    "einops",
    "numpy",
    "transformers",
    "safetensors",
    "accelerate",
    "scipy>=1.14",
]

[project.optional-dependencies]
flashpack = ["flashpack==0.1.2"]
xformers = ["xformers"]


[tool.uv.sources]
xformers = { index = "pytorch" }

[[tool.uv.index]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cu129"
explicit = true

[build-system]
requires = ["uv_build>=0.9.8,<0.10.0"]
build-backend = "uv_build"

[dependency-groups]
dev = [
    "scikit-image>=0.25.2",
]