hSSKrSSKrSSKrSSKJrJr \R "\5r"SS\RR5r "SS5r g)N) MT5ConfigT5Configc@^\rSrSrSrS\\-4U4SjjrSrSr U=r $) T5Encoderz-T5 encoder outputs only the last hidden stateconfigc:>[TU]5 XlX lgN)super__init__encoderr)selfr r __class__s g/var/www/fran/franai/venv/lib/python3.13/site-packages/onnxruntime/transformers/models/t5/t5_encoder.pyr T5Encoder.__init__s   c*URX5S$)Nr)r r input_idsattention_masks rforwardT5Encoder.forwards||I6q99r)rr ) __name__ __module__ __qualname____firstlineno____doc__rrr r__static_attributes__ __classcell__)rs@rrrs!79(< ::rrc j\rSrSrSr\S S\S\S\S\RS\ 4 Sjj5r S \ 4S jr S r g )T5EncoderInputscXlX lgr rrrs rr T5EncoderInputs.__init__s +40>r batch_sizesequence_length vocab_sizedeviceuse_int32_inputsc>U(a[RO[Rn[R"SUS- X4UUS9n[R"X/XSS9nUS:a4[ U5H%n[ R"SUS- 5n SXxSU 24'M' [Xg5$)a Create dummy inputs for T5 encoder. Args: batch_size (int): batch size sequence_length (int): sequence length vocab_size (int): vocabulary size device (torch.device): device of output tensors Returns: T5EncoderInputs: dummy inputs for encoder r)lowhighsizedtyper))r0r)N)torchint32int64randintonesrangerandomr!) r&r'r(r)r*r0rripadding_positions r create_dummyT5EncoderInputs.create_dummy s& 0 U[[MMa.  Z$A^ a :&#)>>!_q5H#I 78"3#3"334'y99rreturnchURUR4Vs/sH ocMUPM nnU$s snfr r$)rv input_lists rto_listT5EncoderInputs.to_listDs3"&..$2E2E!FX!FAa!F XYs//)rrN)F)rrrrr staticmethodintr2r)boolr;listrArrrr!r!s_? "' !:!:!:!: !:  !:!:Frr!) loggingr8r2 transformersrr getLoggerrloggernnModulerr!rGrrrNsA  ,   8 $ : :++r