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

This commit is contained in:
Adil Zouitine
2025-07-28 11:49:05 +02:00
committed by GitHub
parent 98746c7cf9
commit b61a4ded9a

View File

@@ -488,6 +488,8 @@ class PI0FAST(nn.Module):
param.data = param.data.to(dtype=torch_precision)
self.set_requires_grad()
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.padding_side = self.config.padding_side