Spaces:

yizhangliu
/

Grounded-Segment-Anything

Sleeping

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

liuyizhang

add transformers_4_35_0

1ce5e18 8 months ago

No virus

4.26 kB

	# Copyright 2020 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_flax_available,
	is_tf_available,
	is_tokenizers_available,
	is_torch_available,
	)


	_import_structure = {
	"configuration_blenderbot_small": [
	"BLENDERBOT_SMALL_PRETRAINED_CONFIG_ARCHIVE_MAP",
	"BlenderbotSmallConfig",
	"BlenderbotSmallOnnxConfig",
	],
	"tokenization_blenderbot_small": ["BlenderbotSmallTokenizer"],
	}

	try:
	if not is_tokenizers_available():
	raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
	pass
	else:
	_import_structure["tokenization_blenderbot_small_fast"] = ["BlenderbotSmallTokenizerFast"]

	try:
	if not is_torch_available():
	raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
	pass
	else:
	_import_structure["modeling_blenderbot_small"] = [
	"BLENDERBOT_SMALL_PRETRAINED_MODEL_ARCHIVE_LIST",
	"BlenderbotSmallForCausalLM",
	"BlenderbotSmallForConditionalGeneration",
	"BlenderbotSmallModel",
	"BlenderbotSmallPreTrainedModel",
	]

	try:
	if not is_tf_available():
	raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
	pass
	else:
	_import_structure["modeling_tf_blenderbot_small"] = [
	"TFBlenderbotSmallForConditionalGeneration",
	"TFBlenderbotSmallModel",
	"TFBlenderbotSmallPreTrainedModel",
	]

	try:
	if not is_flax_available():
	raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
	pass
	else:
	_import_structure["modeling_flax_blenderbot_small"] = [
	"FlaxBlenderbotSmallForConditionalGeneration",
	"FlaxBlenderbotSmallModel",
	"FlaxBlenderbotSmallPreTrainedModel",
	]

	if TYPE_CHECKING:
	from .configuration_blenderbot_small import (
	BLENDERBOT_SMALL_PRETRAINED_CONFIG_ARCHIVE_MAP,
	BlenderbotSmallConfig,
	BlenderbotSmallOnnxConfig,
	)
	from .tokenization_blenderbot_small import BlenderbotSmallTokenizer

	try:
	if not is_tokenizers_available():
	raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
	pass
	else:
	from .tokenization_blenderbot_small_fast import BlenderbotSmallTokenizerFast

	try:
	if not is_torch_available():
	raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
	pass
	else:
	from .modeling_blenderbot_small import (
	BLENDERBOT_SMALL_PRETRAINED_MODEL_ARCHIVE_LIST,
	BlenderbotSmallForCausalLM,
	BlenderbotSmallForConditionalGeneration,
	BlenderbotSmallModel,
	BlenderbotSmallPreTrainedModel,
	)

	try:
	if not is_tf_available():
	raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
	pass
	else:
	from .modeling_tf_blenderbot_small import (
	TFBlenderbotSmallForConditionalGeneration,
	TFBlenderbotSmallModel,
	TFBlenderbotSmallPreTrainedModel,
	)

	try:
	if not is_flax_available():
	raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
	pass
	else:
	from .modeling_flax_blenderbot_small import (
	FlaxBlenderbotSmallForConditionalGeneration,
	FlaxBlenderbotSmallModel,
	FlaxBlenderbotSmallPreTrainedModel,
	)

	else:
	import sys

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