Update optimum/intel/ipex/modeling_base.py

Co-authored-by: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>

Author: jiqing-feng

optimum/intel/ipex/modeling_base.py

@@ -358,7 +358,8 @@ def _supports_logits_to_keep(self) -> bool:
         Return True if the current model supports the keyword argument `logits_to_keep` in forward()
         to save memory. Checking it in this way allows to avoid using a new model attribute.
         """
-        return "logits_to_keep" in set(inspect.signature(self.model.forward).parameters.keys())
+        logits_to_keep_name = "logits_to_keep" if is_transformers_version(">", "4.49") else "num_logits_to_keep"
+        return logits_to_keep_name in set(inspect.signature(self.model.forward).parameters.keys())
 
     def generate(self, *args, **kwargs):
         if self._add_patch and kwargs.get("assistant_model", None):