We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6f940bc commit 5760936Copy full SHA for 5760936
optimum/intel/ipex/modeling_base.py
@@ -489,7 +489,7 @@ def _supports_logits_to_keep(self) -> bool:
489
def _supports_num_logits_to_keep(self) -> bool:
490
"""
491
Will be deprecated after we no longer support transformers < 4.49
492
-
+
493
Return True if the current model supports the keyword argument `num_logits_to_keep` in forward()
494
to save memory. Checking it in this way allows to avoid using a new model attribute.
495
0 commit comments