admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-21 00:42:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(test): use lerobot model

Browse Source
This commit is contained in:
Steven Palma
2026-03-04 20:59:54 +01:00
parent 3eecf12020
commit ab7dcc2f79
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/pi0_fast/test_pi0_fast_original_vs_lerobot.py
+1 -1
View File
@@ -49,7 +49,7 @@ IMAGE_HEIGHT = 224
IMAGE_WIDTH = 224 IMAGE_WIDTH = 224
NUM_VIEWS = 2 # Number of camera views NUM_VIEWS = 2 # Number of camera views
DEVICE = "cuda" 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 token shape: (batch_size, max_decoding_steps)
EXPECTED_ACTION_TOKENS_SHAPE = (1, 2) EXPECTED_ACTION_TOKENS_SHAPE = (1, 2)