pyproject.toml

[build-system]
requires = ["setuptools>=61"]
build-backend = "setuptools.build_meta"

[project]
name = "prism-video"
version = "0.1.0"
description = "PRISM — Predictive Recomposition via Semantic Latent Decomposition for view-invariant video representation learning"
readme = "README.md"
requires-python = ">=3.10"
# Pinned to the environment the released checkpoint was trained and verified in
# (Python 3.13.13, CUDA 12.8). `uv.lock` freezes the full transitive tree.
dependencies = [
    "torch==2.10.0",
    "transformers==4.57.6",
    "accelerate==1.13.0",       # required by transformers.Trainer
    "huggingface-hub==0.36.2",
    "safetensors==0.7.0",
    "av==17.0.1",               # PyAV — video decoding
    "pillow==12.2.0",
    "numpy==2.2.6",
    "tqdm==4.67.3",
    "openai==2.24.0",           # composer client (talks to a vLLM server)
]

[project.optional-dependencies]
serve = ["vllm==0.17.1"]        # only needed to host the recomposition LLM

[tool.setuptools.packages.find]
include = ["prism*"]