o }oi@s@ddlmZddlmZddlmZdgZGdddeeZdS)) BertModel) BertModule) typecheck BertEncodercs(eZdZdZedfdd ZZS)rzg Wraps around the Huggingface transformers implementation repository for easy use within NeMo. Ncstj|||dd}|S)N) input_idsattention_masktoken_type_idsr)superforward)selfrrrres __class__h/home/ubuntu/.local/lib/python3.10/site-packages/nemo/collections/nlp/modules/common/huggingface/bert.pyr szBertEncoder.forward)NN)__name__ __module__ __qualname____doc__rr __classcell__rrr rrsN) transformersr/nemo.collections.nlp.modules.common.bert_modulernemo.core.classesr__all__rrrrrs