o }oi4@s@ddlmZddlmZddlmZdgZGdddeeZdS))DistilBertModel) BertModule) typecheckDistilBertEncodercs(eZdZdZedfdd ZZS)rzg Wraps around the Huggingface transformers implementation repository for easy use within NeMo. Ncstj||dd}|S)N) input_idsattention_maskr)superforward)selfrrtoken_type_idsres __class__n/home/ubuntu/.local/lib/python3.10/site-packages/nemo/collections/nlp/modules/common/huggingface/distilbert.pyr szDistilBertEncoder.forward)N)__name__ __module__ __qualname____doc__rr __classcell__rrr rrsN) transformersr/nemo.collections.nlp.modules.common.bert_modulernemo.core.classesr__all__rrrrrs