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

[project]
name = "dreamzero"
description = "DreamZero: World Action Models Are Zero-Shot Policies"
version = "1.0.0"
readme = "README.md"
requires-python = "~=3.11,<3.13"
dependencies = [
    "torch>=2.3,<2.7",
    "torchvision>=0.18",
    "transformers==4.51.3",
    "diffusers>=0.30,<0.32",
    "accelerate>=0.27",
    "deepspeed>=0.12,<0.16",
    "hydra-core>=1.3",
    "omegaconf>=2.3",
    "decord",
    "opencv-python>=4.8",
    "av",
    "einops>=0.8",
    "safetensors",
    "sentencepiece",
    "protobuf",
    "zarr",
    "h5py",
    "pyarrow",
    "pandas",
    "scipy",
    "numpy<2",
    "tqdm",
    "bitsandbytes>=0.41",
    "wandb",
    "tensorboard",
]

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