






from typing import Optional

from ..utils.logging import get_logger
from .audio_classificiation import AudioClassification
from .automatic_speech_recognition import AutomaticSpeechRecognition
from .base import TaskTemplate
from .image_classification import ImageClassification
from .language_modeling import LanguageModeling
from .question_answering import QuestionAnsweringExtractive
from .summarization import Summarization
from .text_classification import TextClassification


__UpperCAmelCase      =				[
    "AutomaticSpeechRecognition",
    "AudioClassification",
    "ImageClassification",
    "LanguageModeling",
    "QuestionAnsweringExtractive",
    "Summarization",
    "TaskTemplate",
    "TextClassification",
]

__UpperCAmelCase      =				get_logger(__name__)


__UpperCAmelCase      =				{
    AutomaticSpeechRecognition.task: AutomaticSpeechRecognition,
    AudioClassification.task: AudioClassification,
    ImageClassification.task: ImageClassification,
    LanguageModeling.task: LanguageModeling,
    QuestionAnsweringExtractive.task: QuestionAnsweringExtractive,
    Summarization.task: Summarization,
    TextClassification.task: TextClassification,
}





def 						A_					(						lowercase_			)				->Optional[TaskTemplate]:







 """simple docstring"""

 SCREAMING_SNAKE_CASE				    =		task_template_dict.get('task'			)
 if task_name is None:
  logger.warning(f'''Couldn\'t find template for task \'{task_name}\'. Available templates: {list(lowercase_			)}'''			)
  return None
 SCREAMING_SNAKE_CASE				    =		NAMEaTEMPLATE.get(lowercase_			)
 return template.from_dict(lowercase_			)












import copy

from ...configuration_utils import PretrainedConfig
from ...utils import logging
from ..auto.configuration_auto import AutoConfig


__UpperCAmelCase      =				logging.get_logger(__name__)
class 	a_(		lowercase__	):



     """simple docstring"""



     __snake_case		:				Union[str, Any]  					='''speech-encoder-decoder'''
     __snake_case		:				List[Any]  					=True





     def __init__(		self					:	Any  ,  **lowerCAmelCase__					:	List[Any])							->	Tuple:

      """simple docstring"""
      super().__init__(**lowerCAmelCase__)
      if "encoder" not in kwargs or "decoder" not in kwargs:
       raise ValueError(
           f'''A configuraton of type {self.model_type} cannot be instantiated because not both `encoder` and'''
           f''' `decoder` sub-configurations are passed, but only {kwargs}''')

      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')

      SCREAMING_SNAKE_CASE				    =		AutoConfig.for_model(lowerCAmelCase__  ,  **lowerCAmelCase__)
      SCREAMING_SNAKE_CASE				    =		AutoConfig.for_model(lowerCAmelCase__  ,  **lowerCAmelCase__)
      SCREAMING_SNAKE_CASE				    =		True





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

      """simple docstring"""
      logger.info('Setting `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()  ,  **lowerCAmelCase__)





     def     __UpperCamelCase						(		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




