"docs/vscode:/vscode.git/clone" did not exist on "7032e0203262ebb2ebf55da8d2e01f873973e835"
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:
from .models.mbart import TFMBartForConditionalGeneration, TFMBartModel, TFMBartPreTrainedModel
from .models.mobilebert import (
TF_MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST,
TF_MOBILEVIT_PRETRAINED_MODEL_ARCHIVE_LIST,
TFMobileBertForMaskedLM,
TFMobileBertForMultipleChoice,
TFMobileBertForNextSentencePrediction,
......@@ -6233,6 +6232,9 @@ if TYPE_CHECKING:
TFMobileBertMainLayer,
TFMobileBertModel,
TFMobileBertPreTrainedModel,
)
from .models.mobilevit import (
TF_MOBILEVIT_PRETRAINED_MODEL_ARCHIVE_LIST,
TFMobileViTForImageClassification,
TFMobileViTForSemanticSegmentation,
TFMobileViTModel,
......
......@@ -1647,9 +1647,6 @@ class TFMBartPreTrainedModel(metaclass=DummyObject):
TF_MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST = None
TF_MOBILEVIT_PRETRAINED_MODEL_ARCHIVE_LIST = None
class TFMobileBertForMaskedLM(metaclass=DummyObject):
_backends = ["tf"]
......@@ -1720,6 +1717,9 @@ class TFMobileBertPreTrainedModel(metaclass=DummyObject):
requires_backends(self, ["tf"])
TF_MOBILEVIT_PRETRAINED_MODEL_ARCHIVE_LIST = None
class TFMobileViTForImageClassification(metaclass=DummyObject):
_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