admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-26 22:20:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove previous pi0 and rename pi0_openpi and pi05_openpi

Browse Source
This commit is contained in:
Pepijn
2025-09-22 17:11:29 +02:00
parent 83ed49d9b9
commit 5d9acf9d51
41 changed files with 54 additions and 2813 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/pi0_pi05/test_pi0_openpi.py
+2 -2
View File
@@ -14,7 +14,7 @@ pytestmark = pytest.mark.skipif(
)
from lerobot.policies.factory import make_policy_config # noqa: E402
from lerobot.policies.pi0_openpi import PI0OpenPIConfig, PI0OpenPIPolicy # noqa: E402
from lerobot.policies.pi0 import PI0OpenPIConfig, PI0OpenPIPolicy # noqa: E402
from tests.utils import require_cuda # noqa: E402
@@ -96,7 +96,7 @@ def test_config_creation():
"""Test policy config creation through factory."""
try:
config = make_policy_config(
policy_type="pi0_openpi",
policy_type="pi0",
max_action_dim=7,
max_state_dim=14,
)