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

chore(rewards/robometer): default to lerobot/Robometer-4b model

Browse Source
This commit is contained in:
Khalil Meftah
2026-05-21 14:15:31 +02:00
parent 1b559e1c84
commit a640d752bf
3 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/robometer.mdx
+2 -2
View File
@@ -78,7 +78,7 @@ The model predicts per-frame progress and success internally. The LeRobot reward
from lerobot.rewards.robometer import RobometerConfig, RobometerRewardModel
cfg = RobometerConfig(
pretrained_path="lilkm/Robometer-4B",
pretrained_path="lerobot/Robometer-4B",
device="cuda",
reward_output="progress",
)
@@ -119,7 +119,7 @@ from lerobot.rewards import make_reward_model, make_reward_model_config, make_re
cfg = make_reward_model_config(
"robometer",
pretrained_path="lilkm/Robometer-4B",
pretrained_path="lerobot/Robometer-4B",
device="cuda",
image_key="observation.images.top",
)