From d4e6d60ec3cfd6ad7b7156d5ef5acac24f995ac5 Mon Sep 17 00:00:00 2001 From: Jade Choghari Date: Fri, 28 Nov 2025 10:53:56 +0100 Subject: [PATCH] iterate on cpilot --- src/lerobot/policies/xvla/action_hub.py | 1 - src/lerobot/policies/xvla/configuration_florence2.py | 1 - 2 files changed, 2 deletions(-) diff --git a/src/lerobot/policies/xvla/action_hub.py b/src/lerobot/policies/xvla/action_hub.py index 15c0813ed..7d33e7862 100644 --- a/src/lerobot/policies/xvla/action_hub.py +++ b/src/lerobot/policies/xvla/action_hub.py @@ -280,7 +280,6 @@ class FrankaJoint7ActionSpace(BaseActionSpace): def compute_loss(self, pred, target): assert pred.shape == target.shape, "pred/target shapes must match" - batch_size, seq_len, action_dim = pred.shape joints_loss = self.mse(pred, target) * self.JOINTS_SCALE return {"joints_loss": joints_loss} diff --git a/src/lerobot/policies/xvla/configuration_florence2.py b/src/lerobot/policies/xvla/configuration_florence2.py index 20b32976b..35c006ee0 100644 --- a/src/lerobot/policies/xvla/configuration_florence2.py +++ b/src/lerobot/policies/xvla/configuration_florence2.py @@ -345,7 +345,6 @@ class Florence2Config(PretrainedConfig): if vision_config is not None: vision_config = Florence2VisionConfig(**vision_config) self.vision_config = vision_config - self.vocab_size = self.vocab_size self.text_config = text_config if text_config is not None: