



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

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

__lowerCamelCase         =			frozenset([])

__lowerCamelCase         =			frozenset(['''image'''])

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

__lowerCamelCase         =			frozenset(['''image'''])

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

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

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

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

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

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

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

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

__lowerCamelCase         =			frozenset(['''class_labels'''])

__lowerCamelCase         =			frozenset(['''class_labels'''])

__lowerCamelCase         =			frozenset(['''batch_size'''])

__lowerCamelCase         =			frozenset([])

__lowerCamelCase         =			frozenset(['''batch_size'''])

__lowerCamelCase         =			frozenset([])

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

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

__lowerCamelCase         =			frozenset(['''input_tokens'''])











import logging

from transformers import PretrainedConfig


__lowerCamelCase         =			logging.getLogger(__name__)


__lowerCamelCase         =			{
    '''bertabs-finetuned-cnndm''': '''https://huggingface.co/remi/bertabs-finetuned-cnndm-extractive-abstractive-summarization/resolve/main/config.json''',
}


class   snake_case_ (lowercase__  ):




       """simple docstring"""






       _lowerCamelCase			    =				"""bertabs"""



       def __init__(     self     ,lowercase=30522     ,lowercase=512     ,lowercase=6     ,lowercase=512     ,lowercase=8     ,lowercase=512     ,lowercase=0.2     ,lowercase=6     ,lowercase=768     ,lowercase=8     ,lowercase=2048     ,lowercase=0.2     ,**lowercase     ,):

             """simple docstring"""





             super().__init__(**lowercase)

             UpperCAmelCase_  :	Any				      =		vocab_size
             UpperCAmelCase_  :	int				      =		max_pos

             UpperCAmelCase_  :	List[str]				      =		enc_layers
             UpperCAmelCase_  :	List[str]				      =		enc_hidden_size
             UpperCAmelCase_  :	Any				      =		enc_heads
             UpperCAmelCase_  :	Any				      =		enc_ff_size
             UpperCAmelCase_  :	Optional[int]				      =		enc_dropout

             UpperCAmelCase_  :	List[str]				      =		dec_layers
             UpperCAmelCase_  :	List[Any]				      =		dec_hidden_size
             UpperCAmelCase_  :	List[str]				      =		dec_heads
             UpperCAmelCase_  :	List[Any]				      =		dec_ff_size
             UpperCAmelCase_  :	List[str]				      =		dec_dropout
