o }oi_@s@ddlmZddlmZddlmZdgZGdddeeZdS))GPT2LMHeadModel) GPTModule) typecheck GPT2Encodercs<eZdZdZe          dfdd ZZS)rzg Wraps around the Huggingface transformers implementation repository for easy use within NeMo. NFc s,tj||||||||| | d } |s| S| S)N) input_idsattention_masktoken_type_ids return_dictlabelsoutput_attentionsoutput_hidden_statespast_key_values position_ids use_cache)superforward) selfrrr r r r r rrr max_lengthres __class__h/home/ubuntu/.local/lib/python3.10/site-packages/nemo/collections/nlp/modules/common/huggingface/gpt2.pyrs zGPT2Encoder.forward) NNNFFFNFNr)__name__ __module__ __qualname____doc__rr __classcell__rrrrrsN) transformersr.nemo.collections.nlp.modules.common.gpt_modulernemo.core.classesr__all__rrrrrs