Skip to content

GitLab

"src/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "83fae834c4a09a323ada78a35c08d5b9cb8aa287"
Unverified Commit f5f1e52f authored by Francisco Kurucz's avatar Francisco Kurucz Committed by GitHub
Browse files

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

parent ea5da52e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/transformers/models/bit/modeling_bit.py
View file @ f5f1e52f
...@@ -870,8 +870,8 @@ class BitBackbone(BitPreTrainedModel, BackboneMixin): ...@@ -870,8 +870,8 @@ class BitBackbone(BitPreTrainedModel, BackboneMixin):
>>> url = "http://images.cocodataset.org/val2017/000000039769.jpg" >>> url = "http://images.cocodataset.org/val2017/000000039769.jpg"
>>> image = Image.open(requests.get(url, stream=True).raw) >>> image = Image.open(requests.get(url, stream=True).raw)
>>> processor = AutoImageProcessor.from_pretrained("google/resnetnv2-50") >>> processor = AutoImageProcessor.from_pretrained("google/bit-50")
>>> model = AutoBackbone.from_pretrained("google/resnetnv2-50") >>> model = AutoBackbone.from_pretrained("google/bit-50")
>>> inputs = processor(image, return_tensors="pt") >>> inputs = processor(image, return_tensors="pt")
>>> outputs = model(**inputs) >>> outputs = model(**inputs)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment