Skip to content

Commit d519e22

Browse files
ganteelusenji
authored andcommitted
TF - Fix convnext classification example (huggingface#17261)
1 parent f005bca commit d519e22

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/transformers/models/convnext/modeling_tf_convnext.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -537,7 +537,7 @@ def call(
537537
>>> image = Image.open(requests.get(url, stream=True).raw)
538538
539539
>>> feature_extractor = ConvNextFeatureExtractor.from_pretrained("facebook/convnext-tiny-224")
540-
>>> model = TFViTForImageClassification.from_pretrained("facebook/convnext-tiny-224")
540+
>>> model = TFConvNextForImageClassification.from_pretrained("facebook/convnext-tiny-224")
541541
542542
>>> inputs = feature_extractor(images=image, return_tensors="tf")
543543
>>> outputs = model(**inputs)

0 commit comments

Comments
 (0)