Skip to content

Commit 6fb8863

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

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

optimum/intel/pipelines/pipeline_base.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -155,10 +155,12 @@ def pipeline(
155155
model: Optional[Union[str, "PreTrainedModel"]] = None,
156156
tokenizer: Optional[Union[str, PreTrainedTokenizer, "PreTrainedTokenizerFast"]] = None,
157157
feature_extractor: Optional[Union[str, PreTrainedFeatureExtractor]] = None,
158-
accelerator: Optional[str] = "ipex",
159158
use_fast: bool = True,
160-
torch_dtype=None,
161-
model_kwargs: Dict[str, Any] = None,
159+
token: Optional[Union[str, bool]] = None,
160+
accelerator: Optional[str] = "ort",
161+
revision: Optional[str] = None,
162+
trust_remote_code: Optional[bool] = None,
163+
*model_kwargs,
162164
**kwargs,
163165
) -> Pipeline:
164166
"""

0 commit comments

Comments
 (0)