ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-04 12:51:27 +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

2
tests/policies/pi0_fast/test_pi0_fast_original_vs_lerobot.py
View File

@@ -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)