






'''simple docstring'''



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


SCREAMING_SNAKE_CASE							=						logging.get_logger(__name__)

SCREAMING_SNAKE_CASE							=						{
    'facebook/s2t-wav2vec2-large-en-de': (
        'https://huggingface.co/facebook/s2t-wav2vec2-large-en-de/resolve/main/config.json'
    ),
    # See all Speech2Text models at https://huggingface.co/models?filter=speech2text2
}
class UpperCAmelCase_       (     __A					):




      """simple docstring"""




      UpperCamelCase_             =      '''speech_to_text_2'''
      UpperCamelCase_             =      ['''past_key_values''']
      UpperCamelCase_             =      {'''num_attention_heads''': '''decoder_attention_heads''', '''hidden_size''': '''d_model'''}






      def __init__(			self      :		Tuple  ,   UpperCAmelCase      :		Dict=1_0000  ,   UpperCAmelCase      :		Dict=6  ,   UpperCAmelCase      :		Dict=2048  ,   UpperCAmelCase      :		int=4  ,   UpperCAmelCase      :		Union[str, Any]=0.0  ,   UpperCAmelCase      :		Union[str, Any]=True  ,   UpperCAmelCase      :		Dict="relu"  ,   UpperCAmelCase      :		List[str]=256  ,   UpperCAmelCase      :		List[Any]=0.1  ,   UpperCAmelCase      :		Any=0.0  ,   UpperCAmelCase      :		int=0.0  ,   UpperCAmelCase      :		int=0.0_2  ,   UpperCAmelCase      :		Dict=2  ,   UpperCAmelCase      :		Tuple=True  ,   UpperCAmelCase      :		List[Any]=1  ,   UpperCAmelCase      :		Any=0  ,   UpperCAmelCase      :		Dict=2  ,   UpperCAmelCase      :		str=1024  ,   **UpperCAmelCase      :		Optional[Any]  ,   )	->	Dict:


       '''simple docstring'''

       lowercase       : Dict						=vocab_size
       lowercase       : Optional[Any]						=d_model
       lowercase       : List[str]						=decoder_ffn_dim
       lowercase       : Union[str, Any]						=decoder_layers
       lowercase       : Any						=decoder_attention_heads
       lowercase       : Optional[Any]						=dropout
       lowercase       : Optional[int]						=attention_dropout
       lowercase       : int						=activation_dropout
       lowercase       : List[str]						=activation_function
       lowercase       : List[str]						=init_std
       lowercase       : str						=decoder_layerdrop
       lowercase       : List[Any]						=use_cache
       lowercase       : List[str]						=decoder_layers
       lowercase       : List[str]						=scale_embedding  # scale factor will be sqrt(d_model) if True
       lowercase       : Any						=max_target_positions

       super().__init__(
           pad_token_id=UpperCAmelCase  ,   bos_token_id=UpperCAmelCase  ,   eos_token_id=UpperCAmelCase  ,   decoder_start_token_id=UpperCAmelCase  ,   **UpperCAmelCase  ,   )











'''simple docstring'''



SCREAMING_SNAKE_CASE							=						frozenset(
    [
        'prompt',
        'height',
        'width',
        'guidance_scale',
        'negative_prompt',
        'prompt_embeds',
        'negative_prompt_embeds',
        'cross_attention_kwargs',
    ]
)

SCREAMING_SNAKE_CASE							=						frozenset(['prompt', 'negative_prompt'])

SCREAMING_SNAKE_CASE							=						frozenset([])

SCREAMING_SNAKE_CASE							=						frozenset(['image'])

SCREAMING_SNAKE_CASE							=						frozenset(
    [
        'image',
        'height',
        'width',
        'guidance_scale',
    ]
)

SCREAMING_SNAKE_CASE							=						frozenset(['image'])

SCREAMING_SNAKE_CASE							=						frozenset(
    [
        'prompt',
        'image',
        'height',
        'width',
        'guidance_scale',
        'negative_prompt',
        'prompt_embeds',
        'negative_prompt_embeds',
    ]
)

SCREAMING_SNAKE_CASE							=						frozenset(['prompt', 'image', 'negative_prompt'])

SCREAMING_SNAKE_CASE							=						frozenset(
    [
        # Text guided image variation with an image mask
        'prompt',
        'image',
        'mask_image',
        'height',
        'width',
        'guidance_scale',
        'negative_prompt',
        'prompt_embeds',
        'negative_prompt_embeds',
    ]
)

SCREAMING_SNAKE_CASE							=						frozenset(['prompt', 'image', 'mask_image', 'negative_prompt'])

SCREAMING_SNAKE_CASE							=						frozenset(
    [
        # image variation with an image mask
        'image',
        'mask_image',
        'height',
        'width',
        'guidance_scale',
    ]
)

SCREAMING_SNAKE_CASE							=						frozenset(['image', 'mask_image'])

SCREAMING_SNAKE_CASE							=						frozenset(
    [
        'example_image',
        'image',
        'mask_image',
        'height',
        'width',
        'guidance_scale',
    ]
)

SCREAMING_SNAKE_CASE							=						frozenset(['example_image', 'image', 'mask_image'])

SCREAMING_SNAKE_CASE							=						frozenset(['class_labels'])

SCREAMING_SNAKE_CASE							=						frozenset(['class_labels'])

SCREAMING_SNAKE_CASE							=						frozenset(['batch_size'])

SCREAMING_SNAKE_CASE							=						frozenset([])

SCREAMING_SNAKE_CASE							=						frozenset(['batch_size'])

SCREAMING_SNAKE_CASE							=						frozenset([])

SCREAMING_SNAKE_CASE							=						frozenset(
    [
        'prompt',
        'audio_length_in_s',
        'guidance_scale',
        'negative_prompt',
        'prompt_embeds',
        'negative_prompt_embeds',
        'cross_attention_kwargs',
    ]
)

SCREAMING_SNAKE_CASE							=						frozenset(['prompt', 'negative_prompt'])
SCREAMING_SNAKE_CASE							=						frozenset(['input_tokens'])

SCREAMING_SNAKE_CASE							=						frozenset(['input_tokens'])



