From 6fa78ca8b424ed2ee111ea3d4b7c9a6b7788c327 Mon Sep 17 00:00:00 2001 From: Maximellerbach Date: Mon, 11 May 2026 16:59:14 +0200 Subject: [PATCH] adding deps to pyproject.toml --- pyproject.toml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pyproject.toml b/pyproject.toml index a6785c564..341cd7209 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -216,6 +216,7 @@ topreward = ["lerobot[transformers-dep]"] xvla = ["lerobot[transformers-dep]"] eo1 = ["lerobot[transformers-dep]", "lerobot[qwen-vl-utils-dep]"] hilserl = ["lerobot[transformers-dep]", "lerobot[dataset]", "gym-hil>=0.1.13,<0.2.0", "lerobot[grpcio-dep]", "lerobot[placo-dep]"] +vla_jepa = ["lerobot[transformers-dep]", "lerobot[diffusers-dep]", "lerobot[qwen-vl-utils-dep]"] # Features async = ["lerobot[grpcio-dep]", "lerobot[matplotlib-dep]"] @@ -281,6 +282,7 @@ all = [ # "lerobot[groot]", TODO(Steven): Gr00t requires specific installation instructions for flash-attn "lerobot[xvla]", "lerobot[hilserl]", + "lerobot[vla_jepa]", "lerobot[async]", "lerobot[dev]", "lerobot[test]",