

from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


_UpperCAmelCase		:       List[str]						    =	{
    """configuration_megatron_bert""": ["""MEGATRON_BERT_PRETRAINED_CONFIG_ARCHIVE_MAP""", """MegatronBertConfig"""],
}

try:
					if not is_torch_available():
										raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
					pass
else:
					_UpperCAmelCase		:       Dict						    =	[
					    """MEGATRON_BERT_PRETRAINED_MODEL_ARCHIVE_LIST""",
					    """MegatronBertForCausalLM""",
					    """MegatronBertForMaskedLM""",
					    """MegatronBertForMultipleChoice""",
					    """MegatronBertForNextSentencePrediction""",
					    """MegatronBertForPreTraining""",
					    """MegatronBertForQuestionAnswering""",
					    """MegatronBertForSequenceClassification""",
					    """MegatronBertForTokenClassification""",
					    """MegatronBertModel""",
					    """MegatronBertPreTrainedModel""",
					]

if TYPE_CHECKING:
					from .configuration_megatron_bert import MEGATRON_BERT_PRETRAINED_CONFIG_ARCHIVE_MAP, MegatronBertConfig

					try:
										if not is_torch_available():
															raise OptionalDependencyNotAvailable()
					except OptionalDependencyNotAvailable:
										pass
					else:
										from .modeling_megatron_bert import (
										    MEGATRON_BERT_PRETRAINED_MODEL_ARCHIVE_LIST,
										    MegatronBertForCausalLM,
										    MegatronBertForMaskedLM,
										    MegatronBertForMultipleChoice,
										    MegatronBertForNextSentencePrediction,
										    MegatronBertForPreTraining,
										    MegatronBertForQuestionAnswering,
										    MegatronBertForSequenceClassification,
										    MegatronBertForTokenClassification,
										    MegatronBertModel,
										    MegatronBertPreTrainedModel,
										)

else:
					import sys

					_UpperCAmelCase		:       List[Any]						    =	_LazyModule(__name__, globals()["""__file__"""], _import_structure, module_spec=__spec__)







from ...utils import (
    OptionalDependencyNotAvailable,
    is_torch_available,
    is_transformers_available,
    is_transformers_version,
)


try:
					if not (is_transformers_available() and is_torch_available() and is_transformers_version(""">=""", """4.25.0""")):
										raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
					from ...utils.dummy_torch_and_transformers_objects import UnCLIPImageVariationPipeline, UnCLIPPipeline
else:
					from .pipeline_unclip import UnCLIPPipeline
					from .pipeline_unclip_image_variation import UnCLIPImageVariationPipeline
					from .text_proj import UnCLIPTextProjModel




