hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

chore(pi0fast): TODO comment to warn the need for removal ignore_index (#1593)

Browse Source
This commit is contained in:
Adil Zouitine
2025-07-28 11:49:05 +02:00
committed by GitHub
parent 98746c7cf9
commit b61a4ded9a
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/lerobot/policies/pi0fast/modeling_pi0fast.py
View File

@@ -488,6 +488,8 @@ class PI0FAST(nn.Module):
param.data = param.data.to(dtype=torch_precision) param.data = param.data.to(dtype=torch_precision)
self.set_requires_grad() self.set_requires_grad()
self.image_keys = self.config.image_features.keys() self.image_keys = self.config.image_features.keys()
# TODO: Remove this once we bump transformers to >4.52.0 because the attribute will be removed
# AttributeError: 'PaliGemmaConfig' object has no attribute 'ignore_index'
self.ignore_index = self.pi0_paligemma.config.ignore_index self.ignore_index = self.pi0_paligemma.config.ignore_index
self.padding_side = self.config.padding_side self.padding_side = self.config.padding_side