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

fix push to hub test

Browse Source
This commit is contained in:
Pepijn
2025-09-11 09:18:20 +02:00
parent 514625a7f6
commit 494aa576b2
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_pi0_pi05_hub.py
+2 -1
View File
@@ -84,7 +84,8 @@ def test_hub_loading(model_id="pepijn223/pi0_base_fp32", model_name="PI0"):
print(f" - State dimension: {policy.config.state_dim}") print(f" - State dimension: {policy.config.state_dim}")
print(f" - Action horizon: {policy.config.action_horizon}") print(f" - Action horizon: {policy.config.action_horizon}")
print(f" - Tokenizer max length: {policy.config.tokenizer_max_length}") print(f" - Tokenizer max length: {policy.config.tokenizer_max_length}")
print(f" - discrete_state_input: {policy.config.discrete_state_input}") if model_name == "PI0.5":
print(f" - discrete_state_input: {policy.config.discrete_state_input}")
print(f" - Device: {device}") print(f" - Device: {device}")
print(f" - Dtype: {next(policy.parameters()).dtype}") print(f" - Dtype: {next(policy.parameters()).dtype}")