Skip to content

Commit 6f940bc

Browse files
committed
add comments
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent eb31cd2 commit 6f940bc

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

optimum/intel/ipex/modeling_base.py

+4
Original file line numberDiff line numberDiff line change
@@ -362,6 +362,8 @@ def _supports_logits_to_keep(self) -> bool:
362362

363363
def _supports_num_logits_to_keep(self) -> bool:
364364
"""
365+
Will be deprecated after we no longer support transformers < 4.49
366+
365367
Return True if the current model supports the keyword argument `num_logits_to_keep` in forward()
366368
to save memory. Checking it in this way allows to avoid using a new model attribute.
367369
"""
@@ -486,6 +488,8 @@ def _supports_logits_to_keep(self) -> bool:
486488

487489
def _supports_num_logits_to_keep(self) -> bool:
488490
"""
491+
Will be deprecated after we no longer support transformers < 4.49
492+
489493
Return True if the current model supports the keyword argument `num_logits_to_keep` in forward()
490494
to save memory. Checking it in this way allows to avoid using a new model attribute.
491495
"""

0 commit comments

Comments
 (0)