admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-30 21:19:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix typo (#4048)

Browse Source 
Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
This commit is contained in:
Martino Russi
2026-07-30 15:25:26 +02:00
committed by GitHub
parent 2939168c33
commit 0187856202
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
@@ -150,7 +150,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