Skip to content

Commit

Permalink
Fix documentation references to google/bit-50 model (#32407)
Browse files Browse the repository at this point in the history
  • Loading branch information
JuanFKurucz authored and nbroad1881 committed Aug 7, 2024
1 parent 967fb90 commit b1f619f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/transformers/models/bit/modeling_bit.py
Original file line number Diff line number Diff line change
Expand Up @@ -870,8 +870,8 @@ def forward(
>>> 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)
Expand Down

0 comments on commit b1f619f

Please sign in to comment.