o }oiz @stddlmZddlZddlmZmZddlmZdZdZ dZ dZ d Z d Z d efd d Zdejjd edefddZdS)) is_dataclassN) DictConfig OmegaConf)loggingz;nemo.collections.common.parts.adapter_modules.LinearAdapterzlnemo.collections.asr.parts.submodules.adapters.multi_head_attention_adapter_module.MultiHeadAttentionAdapterzwnemo.collections.asr.parts.submodules.adapters.multi_head_attention_adapter_module.RelPositionMultiHeadAttentionAdapterzlnemo.collections.asr.parts.submodules.adapters.multi_head_attention_adapter_module.PositionalEncodingAdapterzonemo.collections.asr.parts.submodules.adapters.multi_head_attention_adapter_module.RelPositionalEncodingAdapterznemo.collections.asr.parts.submodules.adapters.transformer_multi_head_attention_adapter_module.TransformerMultiHeadAttentionAdaptercfgcCs(t|r t|}t|tst|}|S)N)rr structured isinstancer)rr b/home/ubuntu/.local/lib/python3.10/site-packages/nemo/collections/asr/parts/utils/adapter_utils.py"convert_adapter_cfg_to_dict_config)s   r module module_dimcCst|}gd}d}|D] }||vr|}nq |dur(td|dt|||}||krBtd|jjd|d||}|||<|S)a Update the input dimension of the provided adapter config with some default value. Args: module: The module that implements AdapterModuleMixin. cfg: A DictConfig or a Dataclass representing the adapter config. module_dim: A default module dimension, used if cfg has an incorrect input dimension. Returns: A DictConfig representing the adapter's config. ) in_featuresn_feat hidden_sizeNzKFailed to infer the input dimension of the Adapter cfg. Expected one of : z. Provided config : z Updating z Adapter input dim from z to )r ValueErrorrto_yamlrinfo __class____name__)r rr input_dim_valid_keys input_keykey input_dimr r r update_adapter_cfg_input_dim4s(  r) dataclassesrtorch omegaconfrr nemo.utilsrLINEAR_ADAPTER_CLASSPATHMHA_ADAPTER_CLASSPATHRELMHA_ADAPTER_CLASSPATHPOS_ENCODING_ADAPTER_CLASSPATH"REL_POS_ENCODING_ADAPTER_CLASSPATH!TRANSFORMER_MHA_ADAPTER_CLASSPATHr nnModuleintrr r r r s