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

docs/source/robometer.mdx

@@ -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",
 )

src/lerobot/rewards/robometer/compute_rabc_weights.py

@@ -33,13 +33,13 @@ Usage:
     # Dense per-frame progress for one episode
     python -m lerobot.rewards.robometer.compute_rabc_weights \\
         --dataset-repo-id lerobot/libero_10_image \\
-        --reward-model-path lilkm/Robometer-4B \\
+        --reward-model-path lerobot/Robometer-4B \\
         --episodes 0
 
     # All episodes with batching
     python -m lerobot.rewards.robometer.compute_rabc_weights \\
         --dataset-repo-id lerobot/libero_10_image \\
-        --reward-model-path lilkm/Robometer-4B \\
+        --reward-model-path lerobot/Robometer-4B \\
         --batch-size 16
 """
 
@@ -218,13 +218,13 @@ Examples:
     # Dense per-frame progress for one episode
     python -m lerobot.rewards.robometer.compute_rabc_weights \\
         --dataset-repo-id lerobot/libero_10_image \\
-        --reward-model-path lilkm/Robometer-4B \\
+        --reward-model-path lerobot/Robometer-4B \\
         --episodes 0
 
     # All episodes, smaller batches for memory-constrained GPUs
     python -m lerobot.rewards.robometer.compute_rabc_weights \\
         --dataset-repo-id lerobot/libero_10_image \\
-        --reward-model-path lilkm/Robometer-4B \\
+        --reward-model-path lerobot/Robometer-4B \\
         --batch-size 16
         """,
     )

src/lerobot/rewards/robometer/configuration_robometer.py

@@ -50,7 +50,7 @@ ROBOMETER_SPECIAL_TOKENS = (
 class RobometerConfig(RewardModelConfig):
     """Configuration for the Robometer reward model."""
 
-    pretrained_path: str | None = "lilkm/Robometer-4B"
+    pretrained_path: str | None = "lerobot/Robometer-4B"
     image_key: str = OBS_IMAGES + ".top"
     task_key: str = "task"
     default_task: str | None = None