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

feat: only run pi test on GPU

Browse Source
This commit is contained in:
Pepijn
2025-09-17 15:55:58 +02:00
parent 7aebc526b2
commit 256b0e1e3c
4 changed files with 14 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/test_pi0_original_vs_lerobot.py
+2 -2
View File
@@ -16,7 +16,7 @@ from openpi.models_pytorch.pi0_pytorch import PI0Pytorch
from transformers import AutoTokenizer
from lerobot.policies.pi0_openpi import PI0OpenPIConfig, PI0OpenPIPolicy
from tests.utils import require_nightly_gpu
from tests.utils import require_cuda
DUMMY_ACTION_DIM = 32
DUMMY_STATE_DIM = 32
@@ -318,7 +318,7 @@ def create_original_observation_from_lerobot(lerobot_pi0, batch):
)
@require_nightly_gpu
@require_cuda
def test_pi0_original_vs_lerobot():
"""Test PI0 original implementation vs LeRobot implementation."""
print("Initializing models...")