wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks 

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2025-04-02 08:38:38 +00:00
parent eea7cc424a
commit 0b68f2c2fb
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lerobot/common/policies/pi0fast/modeling_pi0fast.py
View File

@ -846,7 +846,10 @@ class PI0FAST(nn.Module):
decoded_actions = [ decoded_actions = [
torch.tensor( torch.tensor(
self.decode_actions_with_fast( self.decode_actions_with_fast(
tok.tolist(), time_horizon=action_horizon, action_dim=action_dim, relaxed_decoding=self.config.relaxed_decoding tok.tolist(),
time_horizon=action_horizon,
action_dim=action_dim,
relaxed_decoding=self.config.relaxed_decoding,
), ),
device=tokens.device, device=tokens.device,
).squeeze(0) ).squeeze(0)