"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "8f400775fc5bc1011a2674dcfd5408d30d69f678"
Unverified Commit e367a977 authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

Fix MobileVitV2 doctest checkpoint (#24805)

* Fix doctest checkpoint

* Add import torch for mobilevit
parent e5381899
...@@ -1024,9 +1024,10 @@ class MobileViTForSemanticSegmentation(MobileViTPreTrainedModel): ...@@ -1024,9 +1024,10 @@ class MobileViTForSemanticSegmentation(MobileViTPreTrainedModel):
Examples: Examples:
```python ```python
>>> from transformers import AutoImageProcessor, MobileViTForSemanticSegmentation
>>> from PIL import Image
>>> import requests >>> import requests
>>> import torch
>>> from PIL import Image
>>> from transformers import AutoImageProcessor, MobileViTForSemanticSegmentation
>>> 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)
......
...@@ -953,7 +953,6 @@ class MobileViTV2DeepLabV3(nn.Module): ...@@ -953,7 +953,6 @@ class MobileViTV2DeepLabV3(nn.Module):
""", """,
MOBILEVITV2_START_DOCSTRING, MOBILEVITV2_START_DOCSTRING,
) )
# Copied from transformers.models.mobilevit.modeling_mobilevit.MobileViTForSemanticSegmentation with MOBILEVIT->MOBILEVITV2,MobileViT->MobileViTV2,mobilevit->mobilevitv2
class MobileViTV2ForSemanticSegmentation(MobileViTV2PreTrainedModel): class MobileViTV2ForSemanticSegmentation(MobileViTV2PreTrainedModel):
def __init__(self, config: MobileViTV2Config) -> None: def __init__(self, config: MobileViTV2Config) -> None:
super().__init__(config) super().__init__(config)
...@@ -984,15 +983,16 @@ class MobileViTV2ForSemanticSegmentation(MobileViTV2PreTrainedModel): ...@@ -984,15 +983,16 @@ class MobileViTV2ForSemanticSegmentation(MobileViTV2PreTrainedModel):
Examples: Examples:
```python ```python
>>> from transformers import AutoImageProcessor, MobileViTV2ForSemanticSegmentation
>>> from PIL import Image
>>> import requests >>> import requests
>>> import torch
>>> from PIL import Image
>>> from transformers import AutoImageProcessor, MobileViTV2ForSemanticSegmentation
>>> 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)
>>> image_processor = AutoImageProcessor.from_pretrained("apple/deeplabv3-mobilevitv2-small") >>> image_processor = AutoImageProcessor.from_pretrained("apple/mobilevitv2-1.0-imagenet1k-256")
>>> model = MobileViTV2ForSemanticSegmentation.from_pretrained("apple/deeplabv3-mobilevitv2-small") >>> model = MobileViTV2ForSemanticSegmentation.from_pretrained("apple/mobilevitv2-1.0-imagenet1k-256")
>>> inputs = image_processor(images=image, return_tensors="pt") >>> inputs = image_processor(images=image, return_tensors="pt")
......
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