From ab7dcc2f792319d720d190c23d713ca2b2d9fa96 Mon Sep 17 00:00:00 2001 From: Steven Palma Date: Wed, 4 Mar 2026 20:59:54 +0100 Subject: [PATCH] chore(test): use lerobot model --- tests/policies/pi0_fast/test_pi0_fast_original_vs_lerobot.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/policies/pi0_fast/test_pi0_fast_original_vs_lerobot.py b/tests/policies/pi0_fast/test_pi0_fast_original_vs_lerobot.py index 765851f2c..9de781464 100644 --- a/tests/policies/pi0_fast/test_pi0_fast_original_vs_lerobot.py +++ b/tests/policies/pi0_fast/test_pi0_fast_original_vs_lerobot.py @@ -49,7 +49,7 @@ IMAGE_HEIGHT = 224 IMAGE_WIDTH = 224 NUM_VIEWS = 2 # Number of camera views DEVICE = "cuda" -MODEL_PATH_LEROBOT = "jadechoghari/pi0fast-base" +MODEL_PATH_LEROBOT = "lerobot/pi0fast-base" # Expected action token shape: (batch_size, max_decoding_steps) EXPECTED_ACTION_TOKENS_SHAPE = (1, 2)