TF - Fix convnext classification example (#17261)

d3d87b45 · Joao Gante · GitHub · e86faecf · d3d87b45
Unverified Commit d3d87b45 authored May 16, 2022 by Joao Gante Committed by GitHub May 16, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/convnext/modeling_tf_convnext.py src/transformers/models/convnext/modeling_tf_convnext.py +1 -1

No files found.
--- a/src/transformers/models/convnext/modeling_tf_convnext.py
+++ b/src/transformers/models/convnext/modeling_tf_convnext.py
@@ -537,7 +537,7 @@ class TFConvNextForImageClassification(TFConvNextPreTrainedModel, TFSequenceClas
        >>> image = Image.open(requests.get(url, stream=True).raw)

        >>> feature_extractor = ConvNextFeatureExtractor.from_pretrained("facebook/convnext-tiny-224")
-        >>> model = TFViTForImageClassification.from_pretrained("facebook/convnext-tiny-224")
+        >>> model = TFConvNextForImageClassification.from_pretrained("facebook/convnext-tiny-224")

        >>> inputs = feature_extractor(images=image, return_tensors="tf")
        >>> outputs = model(**inputs)