pyproject.toml

[project]
name = "vla-cpp-tooling"
version = "0.3.0"
description = "Python tooling for vla.cpp: HuggingFace -> GGUF converters and the ZeroMQ eval client."
readme = "README.md"
requires-python = ">=3.10"
license = { text = "Apache-2.0" }
dependencies = ["numpy>=1.24"]

[project.optional-dependencies]
# scripts/convert_*_to_gguf.py and the gguf_common / gguf_blocks modules they share
convert = [
    "torch>=2.5",
    "safetensors>=0.4",
    "gguf>=0.10",
    "numpy>=1.24",
]
# eval/client/* talks to vla-server over ZeroMQ. The simulators (LIBERO, SimplerEnv,
# ALOHA) keep their own pinned venvs - see eval/sim/*/setup_*.sh - and are not packaged here.
client = [
    "pyzmq>=25",
    "msgpack>=1.1",
    "msgpack-numpy>=0.4.8",
    "pillow>=10",
    "torch>=2.5",
    # Only used for AutoTokenizer/AutoProcessor.from_pretrained (the pi0 PaliGemma
    # tokenizer). Capped below 5.0: the v5 line is a breaking rewrite we have not
    # tested against.
    "transformers>=4.51,<5",
    "numpy>=1.24",
]

# Deps-only project: the code lives in scripts/ and eval/, not an importable package.
[tool.uv]
package = false