Fix documentation references to google/bit-50 model (#32407)

f5f1e52f · Francisco Kurucz · GitHub · ea5da52e · f5f1e52f
Unverified Commit f5f1e52f authored Aug 05, 2024 by Francisco Kurucz Committed by GitHub Aug 05, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/transformers/models/bit/modeling_bit.py src/transformers/models/bit/modeling_bit.py +2 -2

No files found.
--- a/src/transformers/models/bit/modeling_bit.py
+++ b/src/transformers/models/bit/modeling_bit.py
@@ -870,8 +870,8 @@ class BitBackbone(BitPreTrainedModel, BackboneMixin):
        >>> url = "http://images.cocodataset.org/val2017/000000039769.jpg"
        >>> image = Image.open(requests.get(url, stream=True).raw)

-        >>> processor = AutoImageProcessor.from_pretrained("google/resnetnv2-50")
-        >>> model = AutoBackbone.from_pretrained("google/resnetnv2-50")
+        >>> processor = AutoImageProcessor.from_pretrained("google/bit-50")
+        >>> model = AutoBackbone.from_pretrained("google/bit-50")

        >>> inputs = processor(image, return_tensors="pt")
        >>> outputs = model(**inputs)