fix typo

2026-07-23 17:56:07 +00:00 · 2026-07-17 11:39:14 +02:00
1 changed files with 1 additions and 1 deletions
@@ -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