





import copy

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


_lowerCamelCase 					=      logging.get_logger(__name__)




class 	UpperCamelCase_							(  UpperCamelCase__     ):
							lowerCamelCase_							     =       "encoder-decoder"
							lowerCamelCase_							     =       True


							def __init__(				self     :Optional[int]				,					**__A     :str		) -> int:



									"""simple docstring"""



									super().__init__(**__A		)
									assert (
									    "encoder" in kwargs and "decoder" in kwargs
									), "Config has to be initialized with encoder and decoder config"
									SCREAMING_SNAKE_CASE__								=   kwargs.pop("""encoder"""		)
									SCREAMING_SNAKE_CASE__								=   encoder_config.pop("""model_type"""		)
									SCREAMING_SNAKE_CASE__								=   kwargs.pop("""decoder"""		)
									SCREAMING_SNAKE_CASE__								=   decoder_config.pop("""model_type"""		)

									from ..auto.configuration_auto import AutoConfig

									SCREAMING_SNAKE_CASE__								=   AutoConfig.for_model(__A				,					**__A		)
									SCREAMING_SNAKE_CASE__								=   AutoConfig.for_model(__A				,					**__A		)
									SCREAMING_SNAKE_CASE__								=   True


							@classmethod
							def   _snake_case	(				cls     :str				,					__A     :PretrainedConfig				,					__A     :PretrainedConfig				,					**__A     :List[str]		) -> PretrainedConfig:



									"""simple docstring"""



									logger.info("""Set `config.is_decoder=True` and `config.add_cross_attention=True` for decoder_config"""		)
									SCREAMING_SNAKE_CASE__								=   True
									SCREAMING_SNAKE_CASE__								=   True

									return cls(encoder=encoder_config.to_dict()				,					decoder=decoder_config.to_dict()				,					**__A		)


							def   _snake_case	(				self     :str		) -> Union[str, Any]:



									"""simple docstring"""



									SCREAMING_SNAKE_CASE__								=   copy.deepcopy(self.__dict__		)
									SCREAMING_SNAKE_CASE__								=   self.encoder.to_dict()
									SCREAMING_SNAKE_CASE__								=   self.decoder.to_dict()
									SCREAMING_SNAKE_CASE__								=   self.__class__.model_type
									return output

from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


lowerCamelCase__						=				{
    "configuration_upernet": ["UperNetConfig"],
}

try:
		if not is_torch_available():
				raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
		pass
else:
		lowerCamelCase__						=				[
		    "UperNetForSemanticSegmentation",
		    "UperNetPreTrainedModel",
		]


if TYPE_CHECKING:
		from .configuration_upernet import UperNetConfig

		try:
				if not is_torch_available():
						raise OptionalDependencyNotAvailable()
		except OptionalDependencyNotAvailable:
				pass
		else:
				from .modeling_upernet import UperNetForSemanticSegmentation, UperNetPreTrainedModel


else:
		import sys

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



