We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1e72e85 commit 80c6438Copy full SHA for 80c6438
src/transformers/pipelines/base.py
@@ -798,7 +798,7 @@ class Pipeline(_ScikitCompat, PushToHubMixin):
798
799
def __init__(
800
self,
801
- model: PreTrainedModel,
+ model: "PreTrainedModel",
802
tokenizer: Optional[PreTrainedTokenizer] = None,
803
feature_extractor: Optional[PreTrainedFeatureExtractor] = None,
804
image_processor: Optional[BaseImageProcessor] = None,
0 commit comments