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: