Skip to content

Commit e784dd2

Browse files
Update optimum/intel/pipelines/pipeline_base.py
Co-authored-by: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>
1 parent f401b55 commit e784dd2

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

optimum/intel/pipelines/pipeline_base.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -98,8 +98,10 @@ def load_ipex_model(
9898
model,
9999
targeted_task,
100100
SUPPORTED_TASKS,
101+
subfolder: str = "",
102+
token: Optional[Union[bool, str]] = None,
103+
revision: str = "main",
101104
model_kwargs: Optional[Dict[str, Any]] = None,
102-
hub_kwargs: Optional[Dict[str, Any]] = None,
103105
**kwargs,
104106
):
105107
export = kwargs.pop("export", True)

0 commit comments

Comments
 (0)