U &ºcå,ã@sdZddlZddlmZddlmZmZddlmZm Z ddl m Z m Z ddl mZmZdd lmZmZdd lmZmZdd lmZmZdd lmZdd lmZmZddlmZm Z ddl!m"Z"ddl#m$Z$m%Z%ddl&m'Z'ddl(m)Z)m*Z*ddl+m,Z,m-Z-ddl.m/Z/m0Z0ddl1m2Z2ddl3m4Z4m5Z5ddl6m7Z7m8Z8ddl9m:Z:m;Z;e e?dd„e ee$e/eee:e4e)eeee,e7eefDƒƒZ@ede-fdefdefdefde8fd e"fd!e fd"e'fd#e*fd$efd%e fd&e%fd'e fd(e0fd)e;fd*e5fd+efd,efd-efgƒZAGd.d/„d/ƒZBdS)0z Auto Config class. éN)Ú OrderedDicté)Ú$ALBERT_PRETRAINED_CONFIG_ARCHIVE_MAPÚ AlbertConfig)Ú"BART_PRETRAINED_CONFIG_ARCHIVE_MAPÚ BartConfig)Ú"BERT_PRETRAINED_CONFIG_ARCHIVE_MAPÚ BertConfig)Ú'CAMEMBERT_PRETRAINED_CONFIG_ARCHIVE_MAPÚCamembertConfig)Ú"CTRL_PRETRAINED_CONFIG_ARCHIVE_MAPÚ CTRLConfig)Ú(DISTILBERT_PRETRAINED_CONFIG_ARCHIVE_MAPÚDistilBertConfig)Ú%ELECTRA_PRETRAINED_CONFIG_ARCHIVE_MAPÚ ElectraConfig)ÚEncoderDecoderConfig)Ú&FLAUBERT_PRETRAINED_CONFIG_ARCHIVE_MAPÚFlaubertConfig)Ú"GPT2_PRETRAINED_CONFIG_ARCHIVE_MAPÚ GPT2Config)Ú MarianConfig)Ú(OPENAI_GPT_PRETRAINED_CONFIG_ARCHIVE_MAPÚOpenAIGPTConfig)ÚReformerConfig)Ú%ROBERTA_PRETRAINED_CONFIG_ARCHIVE_MAPÚ RobertaConfig)Ú T5_PRETRAINED_CONFIG_ARCHIVE_MAPÚT5Config)Ú(TRANSFO_XL_PRETRAINED_CONFIG_ARCHIVE_MAPÚTransfoXLConfig)ÚPretrainedConfig)Ú!XLM_PRETRAINED_CONFIG_ARCHIVE_MAPÚ XLMConfig)Ú)XLM_ROBERTA_PRETRAINED_CONFIG_ARCHIVE_MAPÚXLMRobertaConfig)Ú#XLNET_PRETRAINED_CONFIG_ARCHIVE_MAPÚ XLNetConfigccs(|] }| ¡D]\}}||fVqqdS)N)Úitems)Ú.0Zpretrained_mapÚkeyÚvalue©r,úC/tmp/pip-unpacked-wheel-ymerj3tt/transformers/configuration_auto.pyÚ .s ír.Zt5Z distilbertZalbertZ camembertz xlm-robertaZmarianZbartZreformerZrobertaZflaubertZbertz openai-gptZgpt2z transfo-xlZxlnetZxlmZctrlZelectrazencoder-decoderc@s0eZdZdZdd„Zedd„ƒZedd„ƒZdS) Ú AutoConfiga5 :class:`~transformers.AutoConfig` is a generic configuration class that will be instantiated as one of the configuration classes of the library when created with the :func:`~transformers.AutoConfig.from_pretrained` class method. The :func:`~transformers.AutoConfig.from_pretrained` method takes care of returning the correct model class instance based on the `model_type` property of the config object, or when it's missing, falling back to using pattern matching on the `pretrained_model_name_or_path` string. cCs tdƒ‚dS)NzwAutoConfig is designed to be instantiated using the `AutoConfig.from_pretrained(pretrained_model_name_or_path)` method.)ÚEnvironmentError)Úselfr,r,r-Ú__init__jsÿzAutoConfig.__init__cOsFt ¡D]\}}||kr|||ŽSqtd |d t ¡¡¡ƒ‚dS)Nz=Unrecognized model identifier in {}. Should contain one of {}ú, )ÚCONFIG_MAPPINGr(Ú ValueErrorÚformatÚjoinÚkeys)ÚclsÚ model_typeÚargsÚkwargsÚpatternÚ config_classr,r,r-Ú for_modelps ÿÿzAutoConfig.for_modelcKs†tj|fdti|—Ž\}}d|kr`__ for usage. return_unused_kwargs (:obj:`boolean`, optional, defaults to `False`): - If False, then this function returns just the final configuration object. - If True, then this functions returns a tuple `(config, unused_kwargs)` where `unused_kwargs` is a dictionary consisting of the key/value pairs whose keys are not configuration attributes: ie the part of kwargs which has not been used to update `config` and is otherwise ignored. kwargs (:obj:`Dict[str, any]`, optional, defaults to `{}`): key/value pairs with which to update the configuration object after loading. - The values in kwargs of any keys which are configuration attributes will be used to override the loaded values. - Behavior concerning key/value pairs whose keys are *not* configuration attributes is controlled by the `return_unused_kwargs` keyword parameter. Examples:: config = AutoConfig.from_pretrained('bert-base-uncased') # Download configuration from S3 and cache. config = AutoConfig.from_pretrained('./test/bert_saved_model/') # E.g. config (or model) was saved using `save_pretrained('./test/saved_model/')` config = AutoConfig.from_pretrained('./test/bert_saved_model/my_configuration.json') config = AutoConfig.from_pretrained('bert-base-uncased', output_attention=True, foo=False) assert config.output_attention == True config, unused_kwargs = AutoConfig.from_pretrained('bert-base-uncased', output_attention=True, foo=False, return_unused_kwargs=True) assert config.output_attention == True assert unused_kwargs == {'foo': False} Zpretrained_config_archive_mapr:z„Unrecognized model in {}. Should have a `model_type` key in its config.json, or contain one of the following strings in its name: {}r3N) r!Zget_config_dictÚ!ALL_PRETRAINED_CONFIG_ARCHIVE_MAPr4Ú from_dictr(r5r6r7r8)r9Zpretrained_model_name_or_pathr<Z config_dictÚ_r>r=r,r,r-Úfrom_pretrained{s&Fÿÿÿ   þÿzAutoConfig.from_pretrainedN)Ú__name__Ú __module__Ú __qualname__Ú__doc__r2Ú classmethodr?rCr,r,r,r-r/_s   r/)CrGÚloggingÚ collectionsrZconfiguration_albertrrZconfiguration_bartrrZconfiguration_bertrr Zconfiguration_camembertr r Zconfiguration_ctrlr r Zconfiguration_distilbertrrZconfiguration_electrarrZconfiguration_encoder_decoderrZconfiguration_flaubertrrZconfiguration_gpt2rrZconfiguration_marianrZconfiguration_openairrZconfiguration_reformerrZconfiguration_robertarrZconfiguration_t5rrZconfiguration_transfo_xlrr Zconfiguration_utilsr!Zconfiguration_xlmr"r#Zconfiguration_xlm_robertar$r%Zconfiguration_xlnetr&r'Ú getLoggerrDÚloggerÚdictr@r4r/r,r,r,r-Ús‚      ðþíÿ