chore(pi0fast): TODO comment to warn the need for removal ignore_index (#1593)
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user