Skip to content

Commit 40d8c53

Browse files
change "actions_id_pad" to "actions_is_pad"(🐛 Bug) (huggingface#774)
Co-authored-by: Steven Palma <[email protected]>
1 parent 87f1f8a commit 40d8c53

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lerobot/common/policies/pi0/modeling_pi0.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,7 @@ def forward(self, batch: dict[str, Tensor], noise=None, time=None) -> tuple[Tens
313313
state = self.prepare_state(batch)
314314
lang_tokens, lang_masks = self.prepare_language(batch)
315315
actions = self.prepare_action(batch)
316-
actions_is_pad = batch.get("actions_id_pad")
316+
actions_is_pad = batch.get("actions_is_pad")
317317

318318
loss_dict = {}
319319
losses = self.model.forward(images, img_masks, lang_tokens, lang_masks, state, actions, noise, time)

0 commit comments

Comments
 (0)