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


UpperCAmelCase				 =   logging.get_logger(__name__)

UpperCAmelCase				 =   {
    """facebook/opt-125m""": """https://huggingface.co/facebook/opt-125m/blob/main/config.json""",
    """facebook/opt-350m""": """https://huggingface.co/facebook/opt-350m/blob/main/config.json""",
    """facebook/opt-1.3b""": """https://huggingface.co/facebook/opt-1.3b/blob/main/config.json""",
    """facebook/opt-2.7b""": """https://huggingface.co/facebook/opt-2.7b/blob/main/config.json""",
    """facebook/opt-6.7b""": """https://huggingface.co/facebook/opt-6.7b/blob/main/config.json""",
    """facebook/opt-13b""": """https://huggingface.co/facebook/opt-13b/blob/main/config.json""",
}






class 				_SCREAMING_SNAKE_CASE( snake_case__						):
   UpperCamelCase_					:    List[str]			      = 'opt'
   UpperCamelCase_					:    List[str]			      = ['past_key_values']
   def __init__(     self				,  lowerCAmelCase_=5_02_72				,  lowerCAmelCase_=7_68				,  lowerCAmelCase_=12				,  lowerCAmelCase_=30_72				,  lowerCAmelCase_=20_48				,  lowerCAmelCase_=True				,  lowerCAmelCase_=False				,  lowerCAmelCase_=None				,  lowerCAmelCase_=0.1				,  lowerCAmelCase_=0.0				,  lowerCAmelCase_=12				,  lowerCAmelCase_="relu"				,  lowerCAmelCase_=0.0				,  lowerCAmelCase_=0.02				,  lowerCAmelCase_=True				,  lowerCAmelCase_=1				,  lowerCAmelCase_=2				,  lowerCAmelCase_=2				,  lowerCAmelCase_=True				,  lowerCAmelCase_=True				,  **lowerCAmelCase_				,  )						->Any:
        super().__init__(
            pad_token_id=lowerCAmelCase_				,  bos_token_id=lowerCAmelCase_				,  eos_token_id=lowerCAmelCase_				,  **lowerCAmelCase_				,  )
        a	:     Dict						      =    vocab_size
        a	:     int						      =    max_position_embeddings
        a	:     List[Any]						      =    num_attention_heads
        a	:     Union[str, Any]						      =    word_embed_proj_dim if word_embed_proj_dim is not None else hidden_size
        a	:     Any						      =    ffn_dim
        a	:     Optional[int]						      =    hidden_size
        a	:     Optional[Any]						      =    num_hidden_layers
        a	:     Optional[int]						      =    dropout
        a	:     Optional[Any]						      =    attention_dropout
        a	:     Union[str, Any]						      =    activation_function
        a	:     Any						      =    init_std
        a	:     List[Any]						      =    layerdrop
        a	:     str						      =    use_cache
        a	:     Any						      =    do_layer_norm_before
        # We keep these variables at `True` for backward compatibility.
        a	:     List[Any]						      =    enable_bias
        a	:     List[str]						      =    layer_norm_elementwise_affine

        # Note that the only purpose of `_remove_final_layer_norm` is to keep backward compatibility
        # with checkpoints that have been fine-tuned before transformers v4.20.1
        # see https://github.com/facebookresearch/metaseq/pull/164
        a	:     Dict						      =    _remove_final_layer_norm
from typing import List

import datasets
from datasets.tasks import ImageClassification

from ..folder_based_builder import folder_based_builder


UpperCAmelCase				 =   datasets.utils.logging.get_logger(__name__)






class 				_SCREAMING_SNAKE_CASE( folder_based_builder.FolderBasedBuilderConfig						):
   UpperCamelCase_					:    bool			      = None
   UpperCamelCase_					:    bool			      = None






class 				_SCREAMING_SNAKE_CASE( folder_based_builder.FolderBasedBuilder						):
   UpperCamelCase_					:    List[str]			      = datasets.Image()
   UpperCamelCase_					:    Any			      = 'image'
   UpperCamelCase_					:    str			      = ImageFolderConfig
   UpperCamelCase_					:    List[str]  # definition at the bottom of the script
   UpperCamelCase_					:    Union[str, Any]			      = ImageClassification(image_column='image'							,   label_column='label'						)

UpperCAmelCase				 =   [
    """.blp""",
    """.bmp""",
    """.dib""",
    """.bufr""",
    """.cur""",
    """.pcx""",
    """.dcx""",
    """.dds""",
    """.ps""",
    """.eps""",
    """.fit""",
    """.fits""",
    """.fli""",
    """.flc""",
    """.ftc""",
    """.ftu""",
    """.gbr""",
    """.gif""",
    """.grib""",
    """.h5""",
    """.hdf""",
    """.png""",
    """.apng""",
    """.jp2""",
    """.j2k""",
    """.jpc""",
    """.jpf""",
    """.jpx""",
    """.j2c""",
    """.icns""",
    """.ico""",
    """.im""",
    """.iim""",
    """.tif""",
    """.tiff""",
    """.jfif""",
    """.jpe""",
    """.jpg""",
    """.jpeg""",
    """.mpg""",
    """.mpeg""",
    """.msp""",
    """.pcd""",
    """.pxr""",
    """.pbm""",
    """.pgm""",
    """.ppm""",
    """.pnm""",
    """.psd""",
    """.bw""",
    """.rgb""",
    """.rgba""",
    """.sgi""",
    """.ras""",
    """.tga""",
    """.icb""",
    """.vda""",
    """.vst""",
    """.webp""",
    """.wmf""",
    """.emf""",
    """.xbm""",
    """.xpm""",
]
UpperCAmelCase				 =   IMAGE_EXTENSIONS