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

fix: remove unused param

Browse Source
This commit is contained in:
Pepijn
2025-09-12 20:25:55 +02:00
parent 7a03223693
commit d1eefd4e97
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/pi05_openpi/modeling_pi05openpi.py
+1 -1
View File
@@ -1253,7 +1253,7 @@ class PI05OpenPIPolicy(PreTrainedPolicy):
# Action queue logic for n_action_steps > 1 # Action queue logic for n_action_steps > 1
if len(self._action_queue) == 0: if len(self._action_queue) == 0:
actions = self.predict_action_chunk(batch, train=False)[:, : self.config.n_action_steps] actions = self.predict_action_chunk(batch)[:, : self.config.n_action_steps]
# Transpose to get shape (n_action_steps, batch_size, action_dim) # Transpose to get shape (n_action_steps, batch_size, action_dim)
self._action_queue.extend(actions.transpose(0, 1)) self._action_queue.extend(actions.transpose(0, 1))
src/lerobot/policies/pi0_openpi/modeling_pi0openpi.py
+1 -1
View File
@@ -1266,7 +1266,7 @@ class PI0OpenPIPolicy(PreTrainedPolicy):
# Action queue logic for n_action_steps > 1 # Action queue logic for n_action_steps > 1
if len(self._action_queue) == 0: if len(self._action_queue) == 0:
actions = self.predict_action_chunk(batch, train=False)[:, : self.config.n_action_steps] actions = self.predict_action_chunk(batch)[:, : self.config.n_action_steps]
# Transpose to get shape (n_action_steps, batch_size, action_dim) # Transpose to get shape (n_action_steps, batch_size, action_dim)
self._action_queue.extend(actions.transpose(0, 1)) self._action_queue.extend(actions.transpose(0, 1))