This commit is contained in:
Martino Russi
2026-07-17 11:39:14 +02:00
parent 051b13573e
commit cf927acb6e
+1 -1
View File
@@ -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