From cf927acb6eb31044629c0b0f943fffc5294389e1 Mon Sep 17 00:00:00 2001 From: Martino Russi Date: Fri, 17 Jul 2026 11:39:14 +0200 Subject: [PATCH] fix typo --- 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 04e923fdd..6ceb0656c 100644 --- a/src/lerobot/policies/xvla/modeling_xvla.py +++ b/src/lerobot/policies/xvla/modeling_xvla.py @@ -154,7 +154,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