



"""simple docstring"""







from ...configuration_utils import PretrainedConfig
from ...utils import logging


UpperCamelCase     =							logging.get_logger(__name__)

UpperCamelCase     =							{
    """google/realm-cc-news-pretrained-embedder""": (
        """https://huggingface.co/google/realm-cc-news-pretrained-embedder/resolve/main/config.json"""
    ),
    """google/realm-cc-news-pretrained-encoder""": (
        """https://huggingface.co/google/realm-cc-news-pretrained-encoder/resolve/main/config.json"""
    ),
    """google/realm-cc-news-pretrained-scorer""": (
        """https://huggingface.co/google/realm-cc-news-pretrained-scorer/resolve/main/config.json"""
    ),
    """google/realm-cc-news-pretrained-openqa""": (
        """https://huggingface.co/google/realm-cc-news-pretrained-openqa/aresolve/main/config.json"""
    ),
    """google/realm-orqa-nq-openqa""": """https://huggingface.co/google/realm-orqa-nq-openqa/resolve/main/config.json""",
    """google/realm-orqa-nq-reader""": """https://huggingface.co/google/realm-orqa-nq-reader/resolve/main/config.json""",
    """google/realm-orqa-wq-openqa""": """https://huggingface.co/google/realm-orqa-wq-openqa/resolve/main/config.json""",
    """google/realm-orqa-wq-reader""": """https://huggingface.co/google/realm-orqa-wq-reader/resolve/main/config.json""",
    # See all REALM models at https://huggingface.co/models?filter=realm
}






class       UpperCamelCase__			(   _lowerCAmelCase      ):

			"""simple docstring"""


			A__	:					List[Any]							  =     "realm"


			def __init__(      self						,	SCREAMING_SNAKE_CASE__=30522						,	SCREAMING_SNAKE_CASE__=768						,	SCREAMING_SNAKE_CASE__=128						,	SCREAMING_SNAKE_CASE__=12						,	SCREAMING_SNAKE_CASE__=12						,	SCREAMING_SNAKE_CASE__=8						,	SCREAMING_SNAKE_CASE__=3072						,	SCREAMING_SNAKE_CASE__="gelu_new"						,	SCREAMING_SNAKE_CASE__=0.1						,	SCREAMING_SNAKE_CASE__=0.1						,	SCREAMING_SNAKE_CASE__=512						,	SCREAMING_SNAKE_CASE__=2						,	SCREAMING_SNAKE_CASE__=0.0_2						,	SCREAMING_SNAKE_CASE__=1e-12						,	SCREAMING_SNAKE_CASE__=256						,	SCREAMING_SNAKE_CASE__=10						,	SCREAMING_SNAKE_CASE__=1e-3						,	SCREAMING_SNAKE_CASE__=5						,	SCREAMING_SNAKE_CASE__=320						,	SCREAMING_SNAKE_CASE__=13353718						,	SCREAMING_SNAKE_CASE__=5000						,	SCREAMING_SNAKE_CASE__=1						,	SCREAMING_SNAKE_CASE__=0						,	SCREAMING_SNAKE_CASE__=2						,	**SCREAMING_SNAKE_CASE__						,	)      ->	Any:
				super().__init__(pad_token_id=SCREAMING_SNAKE_CASE__						,	bos_token_id=SCREAMING_SNAKE_CASE__						,	eos_token_id=SCREAMING_SNAKE_CASE__						,	**SCREAMING_SNAKE_CASE__		)

				# Common config
				A__											=					vocab_size
				A__											=					max_position_embeddings
				A__											=					hidden_size
				A__											=					retriever_proj_size
				A__											=					num_hidden_layers
				A__											=					num_attention_heads
				A__											=					num_candidates
				A__											=					intermediate_size
				A__											=					hidden_act
				A__											=					hidden_dropout_prob
				A__											=					attention_probs_dropout_prob
				A__											=					initializer_range
				A__											=					type_vocab_size
				A__											=					layer_norm_eps

				# Reader config
				A__											=					span_hidden_size
				A__											=					max_span_width
				A__											=					reader_layer_norm_eps
				A__											=					reader_beam_size
				A__											=					reader_seq_len

				# Retrieval config
				A__											=					num_block_records
				A__											=					searcher_beam_size









"""simple docstring"""







from collections import OrderedDict
from typing import TYPE_CHECKING, Any, Mapping, Optional

from packaging import version

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


if TYPE_CHECKING:
			from ...processing_utils import ProcessorMixin
			from ...utils import TensorType


UpperCamelCase     =							logging.get_logger(__name__)

UpperCamelCase     =							{
    """microsoft/layoutlmv3-base""": """https://huggingface.co/microsoft/layoutlmv3-base/resolve/main/config.json""",
}






class       UpperCamelCase__			(   _lowerCAmelCase      ):

			"""simple docstring"""


			A__	:					int							  =     "layoutlmv3"


			def __init__(      self						,	SCREAMING_SNAKE_CASE__=50265						,	SCREAMING_SNAKE_CASE__=768						,	SCREAMING_SNAKE_CASE__=12						,	SCREAMING_SNAKE_CASE__=12						,	SCREAMING_SNAKE_CASE__=3072						,	SCREAMING_SNAKE_CASE__="gelu"						,	SCREAMING_SNAKE_CASE__=0.1						,	SCREAMING_SNAKE_CASE__=0.1						,	SCREAMING_SNAKE_CASE__=512						,	SCREAMING_SNAKE_CASE__=2						,	SCREAMING_SNAKE_CASE__=0.0_2						,	SCREAMING_SNAKE_CASE__=1e-5						,	SCREAMING_SNAKE_CASE__=1						,	SCREAMING_SNAKE_CASE__=0						,	SCREAMING_SNAKE_CASE__=2						,	SCREAMING_SNAKE_CASE__=1024						,	SCREAMING_SNAKE_CASE__=128						,	SCREAMING_SNAKE_CASE__=128						,	SCREAMING_SNAKE_CASE__=True						,	SCREAMING_SNAKE_CASE__=32						,	SCREAMING_SNAKE_CASE__=128						,	SCREAMING_SNAKE_CASE__=64						,	SCREAMING_SNAKE_CASE__=256						,	SCREAMING_SNAKE_CASE__=True						,	SCREAMING_SNAKE_CASE__=True						,	SCREAMING_SNAKE_CASE__=True						,	SCREAMING_SNAKE_CASE__=224						,	SCREAMING_SNAKE_CASE__=3						,	SCREAMING_SNAKE_CASE__=16						,	SCREAMING_SNAKE_CASE__=None						,	**SCREAMING_SNAKE_CASE__						,	)      ->	Any:
				super().__init__(
				    vocab_size=SCREAMING_SNAKE_CASE__						,	hidden_size=SCREAMING_SNAKE_CASE__						,	num_hidden_layers=SCREAMING_SNAKE_CASE__						,	num_attention_heads=SCREAMING_SNAKE_CASE__						,	intermediate_size=SCREAMING_SNAKE_CASE__						,	hidden_act=SCREAMING_SNAKE_CASE__						,	hidden_dropout_prob=SCREAMING_SNAKE_CASE__						,	attention_probs_dropout_prob=SCREAMING_SNAKE_CASE__						,	max_position_embeddings=SCREAMING_SNAKE_CASE__						,	type_vocab_size=SCREAMING_SNAKE_CASE__						,	initializer_range=SCREAMING_SNAKE_CASE__						,	layer_norm_eps=SCREAMING_SNAKE_CASE__						,	pad_token_id=SCREAMING_SNAKE_CASE__						,	bos_token_id=SCREAMING_SNAKE_CASE__						,	eos_token_id=SCREAMING_SNAKE_CASE__						,	**SCREAMING_SNAKE_CASE__						,	)
				A__											=					max_ad_position_embeddings
				A__											=					coordinate_size
				A__											=					shape_size
				A__											=					has_relative_attention_bias
				A__											=					rel_pos_bins
				A__											=					max_rel_pos
				A__											=					has_spatial_attention_bias
				A__											=					rel_ad_pos_bins
				A__											=					max_rel_ad_pos
				A__											=					text_embed
				A__											=					visual_embed
				A__											=					input_size
				A__											=					num_channels
				A__											=					patch_size
				A__											=					classifier_dropout






class       UpperCamelCase__			(   _lowerCAmelCase      ):

			"""simple docstring"""


			A__	:					Union[str, Any]							  =     version.parse("1.12"      )


			@property
			def   snake_case__	(      self		)      ->	Mapping[str, Mapping[int, str]]:
				# The order of inputs is different for question answering and sequence classification
				if self.task in ["question-answering", "sequence-classification"]:
					return OrderedDict(
					    [
					        ("input_ids", {0: "batch", 1: "sequence"}),
					        ("attention_mask", {0: "batch", 1: "sequence"}),
					        ("bbox", {0: "batch", 1: "sequence"}),
					        ("pixel_values", {0: "batch", 1: "num_channels", 2: "height", 3: "width"}),
					    ]		)
				else:
					return OrderedDict(
					    [
					        ("input_ids", {0: "batch", 1: "sequence"}),
					        ("bbox", {0: "batch", 1: "sequence"}),
					        ("attention_mask", {0: "batch", 1: "sequence"}),
					        ("pixel_values", {0: "batch", 1: "num_channels"}),
					    ]		)


			@property
			def   snake_case__	(      self		)      ->	float:
				return 1e-5


			@property
			def   snake_case__	(      self		)      ->	int:
				return 12







			def   snake_case__	(      self						,	SCREAMING_SNAKE_CASE__						,	SCREAMING_SNAKE_CASE__ = -1						,	SCREAMING_SNAKE_CASE__ = -1						,	SCREAMING_SNAKE_CASE__ = False						,	SCREAMING_SNAKE_CASE__ = None						,	SCREAMING_SNAKE_CASE__ = 3						,	SCREAMING_SNAKE_CASE__ = 40						,	SCREAMING_SNAKE_CASE__ = 40						,	)      ->	Mapping[str, Any]:
				setattr(processor.image_processor						,	"apply_ocr"						,	SCREAMING_SNAKE_CASE__		)

				# If dynamic axis (-1) we forward with a fixed dimension of 2 samples to avoid optimizations made by ONNX
				A__											=					compute_effective_axis_dimension(
				    SCREAMING_SNAKE_CASE__						,	fixed_dimension=OnnxConfig.default_fixed_batch						,	num_token_to_add=0		)
				# If dynamic axis (-1) we forward with a fixed dimension of 8 tokens to avoid optimizations made by ONNX
				A__											=					processor.tokenizer.num_special_tokens_to_add(SCREAMING_SNAKE_CASE__		)
				A__											=					compute_effective_axis_dimension(
				    SCREAMING_SNAKE_CASE__						,	fixed_dimension=OnnxConfig.default_fixed_sequence						,	num_token_to_add=SCREAMING_SNAKE_CASE__		)
				# Generate dummy inputs according to compute batch and sequence
				A__											=					[[" ".join([processor.tokenizer.unk_token]		) * seq_length]] * batch_size

				# Generate dummy bounding boxes
				A__											=					[[[48, 84, 73, 128]]] * batch_size

				# If dynamic axis (-1) we forward with a fixed dimension of 2 samples to avoid optimizations made by ONNX
				# batch_size = compute_effective_axis_dimension(batch_size, fixed_dimension=OnnxConfig.default_fixed_batch)
				A__											=					self._generate_dummy_images(SCREAMING_SNAKE_CASE__						,	SCREAMING_SNAKE_CASE__						,	SCREAMING_SNAKE_CASE__						,	SCREAMING_SNAKE_CASE__		)

				A__											=					dict(
				    processor(
				        SCREAMING_SNAKE_CASE__						,	text=SCREAMING_SNAKE_CASE__						,	boxes=SCREAMING_SNAKE_CASE__						,	return_tensors=SCREAMING_SNAKE_CASE__						,	)		)

				return inputs




