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

add preprocess tests

Browse Source
This commit is contained in:
Pepijn
2025-09-12 21:41:25 +02:00
parent 376cc772ff
commit c8163662ad
3 changed files with 130 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/pi05_openpi/modeling_pi05openpi.py
-1
View File
@@ -961,7 +961,6 @@ class PI05OpenPIPolicy(PreTrainedPolicy):
except Exception as e:
print(f"Warning: Could not remap state dict keys: {e}")
print("Returning model without loading pretrained weights")
return model
src/lerobot/policies/pi0_openpi/modeling_pi0openpi.py
-1
View File
@@ -978,7 +978,6 @@ class PI0OpenPIPolicy(PreTrainedPolicy):
except Exception as e:
print(f"Warning: Could not remap state dict keys: {e}")
print("Returning model without loading pretrained weights")
return model