
"""simple docstring"""


from collections import OrderedDict
from typing import Mapping

from ...configuration_utils import PretrainedConfig
from ...onnx import OnnxConfig
from ...utils import logging


A_						       =				logging.get_logger(__name__)

A_						       =				{
    """nghuyong/ernie-1.0-base-zh""": """https://huggingface.co/nghuyong/ernie-1.0-base-zh/resolve/main/config.json""",
    """nghuyong/ernie-2.0-base-en""": """https://huggingface.co/nghuyong/ernie-2.0-base-en/resolve/main/config.json""",
    """nghuyong/ernie-2.0-large-en""": """https://huggingface.co/nghuyong/ernie-2.0-large-en/resolve/main/config.json""",
    """nghuyong/ernie-3.0-base-zh""": """https://huggingface.co/nghuyong/ernie-3.0-base-zh/resolve/main/config.json""",
    """nghuyong/ernie-3.0-medium-zh""": """https://huggingface.co/nghuyong/ernie-3.0-medium-zh/resolve/main/config.json""",
    """nghuyong/ernie-3.0-mini-zh""": """https://huggingface.co/nghuyong/ernie-3.0-mini-zh/resolve/main/config.json""",
    """nghuyong/ernie-3.0-micro-zh""": """https://huggingface.co/nghuyong/ernie-3.0-micro-zh/resolve/main/config.json""",
    """nghuyong/ernie-3.0-nano-zh""": """https://huggingface.co/nghuyong/ernie-3.0-nano-zh/resolve/main/config.json""",
    """nghuyong/ernie-gram-zh""": """https://huggingface.co/nghuyong/ernie-gram-zh/resolve/main/config.json""",
    """nghuyong/ernie-health-zh""": """https://huggingface.co/nghuyong/ernie-health-zh/resolve/main/config.json""",
}




class    __lowerCamelCase     (   lowerCAmelCase ):
				a__:			int									=						'ernie'




				def __init__(							self       ,	UpperCAmelCase=3_0522       ,	UpperCAmelCase=768       ,	UpperCAmelCase=12       ,	UpperCAmelCase=12       ,	UpperCAmelCase=3072       ,	UpperCAmelCase="gelu"       ,	UpperCAmelCase=0.1       ,	UpperCAmelCase=0.1       ,	UpperCAmelCase=512       ,	UpperCAmelCase=2       ,	UpperCAmelCase=3       ,	UpperCAmelCase=False       ,	UpperCAmelCase=0.0_2       ,	UpperCAmelCase=1e-1_2       ,	UpperCAmelCase=0       ,	UpperCAmelCase="absolute"       ,	UpperCAmelCase=True       ,	UpperCAmelCase=None       ,	**UpperCAmelCase       ,	):
							super().__init__(pad_token_id=UpperCAmelCase       ,	**UpperCAmelCase      )

							lowerCamelCase_ 		=     vocab_size
							lowerCamelCase_ 		=     hidden_size
							lowerCamelCase_ 		=     num_hidden_layers
							lowerCamelCase_ 		=     num_attention_heads
							lowerCamelCase_ 		=     hidden_act
							lowerCamelCase_ 		=     intermediate_size
							lowerCamelCase_ 		=     hidden_dropout_prob
							lowerCamelCase_ 		=     attention_probs_dropout_prob
							lowerCamelCase_ 		=     max_position_embeddings
							lowerCamelCase_ 		=     type_vocab_size
							lowerCamelCase_ 		=     task_type_vocab_size
							lowerCamelCase_ 		=     use_task_id
							lowerCamelCase_ 		=     initializer_range
							lowerCamelCase_ 		=     layer_norm_eps
							lowerCamelCase_ 		=     position_embedding_type
							lowerCamelCase_ 		=     use_cache
							lowerCamelCase_ 		=     classifier_dropout




class    __lowerCamelCase     (   lowerCAmelCase ):




				@property
				def 					UpperCAmelCase__	(							self      ):
							if self.task == "multiple-choice":
										lowerCamelCase_ 		=     {0: '''batch''', 1: '''choice''', 2: '''sequence'''}
							else:
										lowerCamelCase_ 		=     {0: '''batch''', 1: '''sequence'''}
							return OrderedDict(
							    [
							        ('''input_ids''', dynamic_axis),
							        ('''attention_mask''', dynamic_axis),
							        ('''token_type_ids''', dynamic_axis),
							        ('''task_type_ids''', dynamic_axis),
							    ]      )





"""simple docstring"""


from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available, is_vision_available


A_						       =				{
    """configuration_flava""": [
        """FLAVA_PRETRAINED_CONFIG_ARCHIVE_MAP""",
        """FlavaConfig""",
        """FlavaImageCodebookConfig""",
        """FlavaImageConfig""",
        """FlavaMultimodalConfig""",
        """FlavaTextConfig""",
    ],
}

try:
						if not is_vision_available():
												raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
						pass
else:
						A_						       =				["""FlavaFeatureExtractor"""]
						A_						       =				["""FlavaImageProcessor"""]
						A_						       =				["""FlavaProcessor"""]

try:
						if not is_torch_available():
												raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
						pass
else:
						A_						       =				[
						    """FLAVA_PRETRAINED_MODEL_ARCHIVE_LIST""",
						    """FlavaForPreTraining""",
						    """FlavaImageCodebook""",
						    """FlavaImageModel""",
						    """FlavaModel""",
						    """FlavaMultimodalModel""",
						    """FlavaPreTrainedModel""",
						    """FlavaTextModel""",
						]

if TYPE_CHECKING:
						from .configuration_flava import (
						    FLAVA_PRETRAINED_CONFIG_ARCHIVE_MAP,
						    FlavaConfig,
						    FlavaImageCodebookConfig,
						    FlavaImageConfig,
						    FlavaMultimodalConfig,
						    FlavaTextConfig,
						)

						try:
												if not is_vision_available():
																		raise OptionalDependencyNotAvailable()
						except OptionalDependencyNotAvailable:
												pass
						else:
												from .feature_extraction_flava import FlavaFeatureExtractor
												from .image_processing_flava import FlavaImageProcessor
												from .processing_flava import FlavaProcessor

						try:
												if not is_torch_available():
																		raise OptionalDependencyNotAvailable()
						except OptionalDependencyNotAvailable:
												pass
						else:
												from .modeling_flava import (
												    FLAVA_PRETRAINED_MODEL_ARCHIVE_LIST,
												    FlavaForPreTraining,
												    FlavaImageCodebook,
												    FlavaImageModel,
												    FlavaModel,
												    FlavaMultimodalModel,
												    FlavaPreTrainedModel,
												    FlavaTextModel,
												)

else:
						import sys

						A_						       =				_LazyModule(__name__, globals()["""__file__"""], _import_structure, module_spec=__spec__)



