add type hint in pipeline model argument (#23740)

* add type hint in pipeline model argument * add pretrainedmodel and tfpretainedmodel type hint * make type hints string

add type hint in pipeline model argument (#23740)
* add type hint in pipeline model argument * add pretrainedmodel and tfpretainedmodel type hint * make type hints string
af45ec0a · Samin Yasar · GitHub · 4b6a5a7c · af45ec0a
Unverified Commit af45ec0a authored May 30, 2023 by Samin Yasar Committed by GitHub May 30, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/pipelines/__init__.py src/transformers/pipelines/__init__.py +1 -1

No files found.
--- a/src/transformers/pipelines/__init__.py
+++ b/src/transformers/pipelines/__init__.py
@@ -505,7 +505,7 @@ def clean_custom_task(task_info):
 def pipeline(
    task: str = None,
-    model: Optional = None,
+    model: Optional[Union[str, "PreTrainedModel", "TFPreTrainedModel"]] = None,
    config: Optional[Union[str, PretrainedConfig]] = None,
    tokenizer: Optional[Union[str, PreTrainedTokenizer, "PreTrainedTokenizerFast"]] = None,
    feature_extractor: Optional[Union[str, PreTrainedFeatureExtractor]] = None,