o }oi@s@ddlmZddlmZddlmZdgZGdddeeZdS)) AlbertModel) BertModule) typecheck AlbertEncodercs&eZdZdZefddZZS)rzg Wraps around the Huggingface transformers implementation repository for easy use within NeMo. cstj|||dd}|S)N) input_idsattention_masktoken_type_idsr)superforward)selfrrrres __class__j/home/ubuntu/.local/lib/python3.10/site-packages/nemo/collections/nlp/modules/common/huggingface/albert.pyr szAlbertEncoder.forward)__name__ __module__ __qualname____doc__rr __classcell__rrr rrsN) transformersr/nemo.collections.nlp.modules.common.bert_modulernemo.core.classesr__all__rrrrrs