admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 10:16:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix typo

Browse Source
This commit is contained in:
Martino Russi
2026-07-17 11:39:14 +02:00
parent 051b13573e
commit cf927acb6e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/xvla/modeling_xvla.py
+1 -1
View File
@@ -154,7 +154,7 @@ class XVLAModel(nn.Module):
# Freeze or unfreeze policy transformer # Freeze or unfreeze policy transformer
if not self.config.train_policy_transformer: if not self.config.train_policy_transformer:
for name, param in self.transformer.named_parameters(): for name, param in self.transformer.named_parameters():
if "soft_prompts" not in name: if "soft_prompt" not in name:
param.requires_grad = False param.requires_grad = False
# Freeze or unfreeze soft prompts # Freeze or unfreeze soft prompts