From 0187856202edaacd5a6a2fb1668b68c8d09d5274 Mon Sep 17 00:00:00 2001 From: Martino Russi <77496684+nepyope@users.noreply.github.com> Date: Thu, 30 Jul 2026 15:25:26 +0200 Subject: [PATCH] fix typo (#4048) Co-authored-by: Steven Palma --- src/lerobot/policies/xvla/modeling_xvla.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/policies/xvla/modeling_xvla.py b/src/lerobot/policies/xvla/modeling_xvla.py index 4b5e85d43..c58eefc6f 100644 --- a/src/lerobot/policies/xvla/modeling_xvla.py +++ b/src/lerobot/policies/xvla/modeling_xvla.py @@ -150,7 +150,7 @@ class XVLAModel(nn.Module): # Freeze or unfreeze policy transformer if not self.config.train_policy_transformer: for name, param in self.transformer.named_parameters(): - if "soft_prompts" not in name: + if "soft_prompt" not in name: param.requires_grad = False # Freeze or unfreeze soft prompts