Skip to content

Commit b058d11

Browse files
committed
fix typo
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent d9af9fe commit b058d11

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

optimum/intel/ipex/modeling_base.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -358,7 +358,7 @@ def _supports_logits_to_keep(self) -> bool:
358358
Return True if the current model supports the keyword argument `logits_to_keep` in forward()
359359
to save memory. Checking it in this way allows to avoid using a new model attribute.
360360
"""
361-
return "logits_to_keep" in set(inspect.signature(self.forward).parameters.keys())
361+
return "logits_to_keep" in set(inspect.signature(self.model.forward).parameters.keys())
362362

363363
def generate(self, *args, **kwargs):
364364
if self._add_patch and kwargs.get("assistant_model", None):
@@ -475,7 +475,7 @@ def _supports_logits_to_keep(self) -> bool:
475475
Return True if the current model supports the keyword argument `logits_to_keep` in forward()
476476
to save memory. Checking it in this way allows to avoid using a new model attribute.
477477
"""
478-
return "logits_to_keep" in set(inspect.signature(self.forward).parameters.keys())
478+
return "logits_to_keep" in set(inspect.signature(self.model.forward).parameters.keys())
479479

480480
def _init_warmup(self):
481481
inputs = prepare_jit_inputs(self.model, self.export_feature, False)

0 commit comments

Comments
 (0)