diff --git a/optimum/intel/ipex/modeling_base.py b/optimum/intel/ipex/modeling_base.py index 5a53445469..b9f9a708da 100644 --- a/optimum/intel/ipex/modeling_base.py +++ b/optimum/intel/ipex/modeling_base.py @@ -232,6 +232,7 @@ def _from_pretrained( model = TasksManager.get_model_from_task( task, model_id, + library_name="transformers", trust_remote_code=trust_remote_code, torch_dtype=torch_dtype, _commit_hash=commit_hash, @@ -267,6 +268,7 @@ def forward( input_ids: torch.Tensor, attention_mask: torch.Tensor, token_type_ids: torch.Tensor = None, + position_ids: torch.Tensor = None, **kwargs, ): inputs = { @@ -277,6 +279,9 @@ def forward( if "token_type_ids" in self.input_names: inputs["token_type_ids"] = token_type_ids + if "position_ids" in self.input_names: + inputs["position_ids"] = position_ids + outputs = self._call_model(**inputs) if isinstance(outputs, dict): model_output = ModelOutput(**outputs)