diff --git a/README.md b/README.md index d73b1c6..69a5206 100644 --- a/README.md +++ b/README.md @@ -69,7 +69,10 @@ pip install flashinfer-python -i https://flashinfer.ai/whl/cu128/torch2.9/ > For other CUDA/PyTorch combinations, see [FlashInfer installation](https://docs.flashinfer.ai/installation.html). > If FlashInfer is not installed, the model falls back to SDPA (PyTorch native attention) via `--use_sdpa`. -**5. Visualization dependencies (optional)** +**5. Optional export and sky-mask dependencies** + +The browser point-cloud viewer is installed with the base package. Install the +optional `vis` extra only if you need GLB export or sky segmentation. ```bash pip install -e ".[vis]" diff --git a/pyproject.toml b/pyproject.toml index 29c6fb8..70e7f68 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -12,10 +12,12 @@ dependencies = [ "tqdm", "scipy", "torchvision", + "matplotlib", + "viser>=0.2.23", ] [project.optional-dependencies] -vis = ["viser>=0.2.23", "trimesh", "matplotlib", "onnxruntime", "requests"] +vis = ["trimesh", "onnxruntime", "requests"] demo = ["lingbot-map[vis]"] [build-system]