Unverified Commit 499770c0 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix imports of TF MobileViT (#22065)

* Fix imports of TF MobileViT

* Fix copies
parent bdec2768
...@@ -6222,7 +6222,6 @@ if TYPE_CHECKING: ...@@ -6222,7 +6222,6 @@ if TYPE_CHECKING:
from .models.mbart import TFMBartForConditionalGeneration, TFMBartModel, TFMBartPreTrainedModel from .models.mbart import TFMBartForConditionalGeneration, TFMBartModel, TFMBartPreTrainedModel
from .models.mobilebert import ( from .models.mobilebert import (
TF_MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST, TF_MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST,
TF_MOBILEVIT_PRETRAINED_MODEL_ARCHIVE_LIST,
TFMobileBertForMaskedLM, TFMobileBertForMaskedLM,
TFMobileBertForMultipleChoice, TFMobileBertForMultipleChoice,
TFMobileBertForNextSentencePrediction, TFMobileBertForNextSentencePrediction,
...@@ -6233,6 +6232,9 @@ if TYPE_CHECKING: ...@@ -6233,6 +6232,9 @@ if TYPE_CHECKING:
TFMobileBertMainLayer, TFMobileBertMainLayer,
TFMobileBertModel, TFMobileBertModel,
TFMobileBertPreTrainedModel, TFMobileBertPreTrainedModel,
)
from .models.mobilevit import (
TF_MOBILEVIT_PRETRAINED_MODEL_ARCHIVE_LIST,
TFMobileViTForImageClassification, TFMobileViTForImageClassification,
TFMobileViTForSemanticSegmentation, TFMobileViTForSemanticSegmentation,
TFMobileViTModel, TFMobileViTModel,
......
...@@ -1647,9 +1647,6 @@ class TFMBartPreTrainedModel(metaclass=DummyObject): ...@@ -1647,9 +1647,6 @@ class TFMBartPreTrainedModel(metaclass=DummyObject):
TF_MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
TF_MOBILEVIT_PRETRAINED_MODEL_ARCHIVE_LIST = None
class TFMobileBertForMaskedLM(metaclass=DummyObject): class TFMobileBertForMaskedLM(metaclass=DummyObject):
_backends = ["tf"] _backends = ["tf"]
...@@ -1720,6 +1717,9 @@ class TFMobileBertPreTrainedModel(metaclass=DummyObject): ...@@ -1720,6 +1717,9 @@ class TFMobileBertPreTrainedModel(metaclass=DummyObject):
requires_backends(self, ["tf"]) requires_backends(self, ["tf"])
TF_MOBILEVIT_PRETRAINED_MODEL_ARCHIVE_LIST = None
class TFMobileViTForImageClassification(metaclass=DummyObject): class TFMobileViTForImageClassification(metaclass=DummyObject):
_backends = ["tf"] _backends = ["tf"]
......
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