Spaces:

yizhangliu
/

Grounded-Segment-Anything

Sleeping

Grounded-Segment-Anything / transformers_4_35_0 /models /deit /__init__.py

liuyizhang

add transformers_4_35_0

1ce5e18 8 months ago

No virus

3.49 kB

	# Copyright 2021 The HuggingFace Team. All rights reserved.
	#
	# Licensed under the Apache License, Version 2.0 (the "License");
	# you may not use this file except in compliance with the License.
	# You may obtain a copy of the License at
	#
	# http://www.apache.org/licenses/LICENSE-2.0
	#
	# Unless required by applicable law or agreed to in writing, software
	# distributed under the License is distributed on an "AS IS" BASIS,
	# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	# See the License for the specific language governing permissions and
	# limitations under the License.
	from typing import TYPE_CHECKING

	from ...utils import (
	OptionalDependencyNotAvailable,
	_LazyModule,
	is_tf_available,
	is_torch_available,
	is_vision_available,
	)


	_import_structure = {"configuration_deit": ["DEIT_PRETRAINED_CONFIG_ARCHIVE_MAP", "DeiTConfig", "DeiTOnnxConfig"]}

	try:
	if not is_vision_available():
	raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
	pass
	else:
	_import_structure["feature_extraction_deit"] = ["DeiTFeatureExtractor"]
	_import_structure["image_processing_deit"] = ["DeiTImageProcessor"]

	try:
	if not is_torch_available():
	raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
	pass
	else:
	_import_structure["modeling_deit"] = [
	"DEIT_PRETRAINED_MODEL_ARCHIVE_LIST",
	"DeiTForImageClassification",
	"DeiTForImageClassificationWithTeacher",
	"DeiTForMaskedImageModeling",
	"DeiTModel",
	"DeiTPreTrainedModel",
	]

	try:
	if not is_tf_available():
	raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
	pass
	else:
	_import_structure["modeling_tf_deit"] = [
	"TF_DEIT_PRETRAINED_MODEL_ARCHIVE_LIST",
	"TFDeiTForImageClassification",
	"TFDeiTForImageClassificationWithTeacher",
	"TFDeiTForMaskedImageModeling",
	"TFDeiTModel",
	"TFDeiTPreTrainedModel",
	]


	if TYPE_CHECKING:
	from .configuration_deit import DEIT_PRETRAINED_CONFIG_ARCHIVE_MAP, DeiTConfig, DeiTOnnxConfig

	try:
	if not is_vision_available():
	raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
	pass
	else:
	from .feature_extraction_deit import DeiTFeatureExtractor
	from .image_processing_deit import DeiTImageProcessor

	try:
	if not is_torch_available():
	raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
	pass
	else:
	from .modeling_deit import (
	DEIT_PRETRAINED_MODEL_ARCHIVE_LIST,
	DeiTForImageClassification,
	DeiTForImageClassificationWithTeacher,
	DeiTForMaskedImageModeling,
	DeiTModel,
	DeiTPreTrainedModel,
	)

	try:
	if not is_tf_available():
	raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
	pass
	else:
	from .modeling_tf_deit import (
	TF_DEIT_PRETRAINED_MODEL_ARCHIVE_LIST,
	TFDeiTForImageClassification,
	TFDeiTForImageClassificationWithTeacher,
	TFDeiTForMaskedImageModeling,
	TFDeiTModel,
	TFDeiTPreTrainedModel,
	)


	else:
	import sys

	sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)