admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-25 02:36:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

linting

Browse Source
This commit is contained in:
Maximellerbach
2026-06-09 17:12:50 +02:00
parent 31ddb8f493
commit c1332ac37e
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/vla_jepa/modeling_vla_jepa.py
+1 -3
View File
@@ -174,9 +174,7 @@ class VLAJEPAModel(nn.Module):
embodied_idx = (input_ids == self.embodied_action_token_id).nonzero(as_tuple=True) embodied_idx = (input_ids == self.embodied_action_token_id).nonzero(as_tuple=True)
action_idx = None action_idx = None
if need_action_tokens: if need_action_tokens:
action_mask = torch.isin( action_mask = torch.isin(input_ids, torch.tensor(self.action_token_ids, device=input_ids.device))
input_ids, torch.tensor(self.action_token_ids, device=input_ids.device)
)
action_idx = action_mask.nonzero(as_tuple=True) action_idx = action_mask.nonzero(as_tuple=True)
device_type = next(self.parameters()).device.type device_type = next(self.parameters()).device.type