`isddlZddlZddlZddlmZmZddlmZddlm Z m Z ddl m Z m Z mZmZmZddlZddlmZddlmZmZmZmZd d lmZmZmZe rdd lmZdd lm Z dd l!m"Z"ddl#m$Z$erddl%m&Z&ej'e(Z)dZ*dZ+ej,GddZ-GddeZ.Gdde.eZ/Gdde/Z0dS)N)ABCabstractmethod) OrderedDict)IterableMapping) TYPE_CHECKINGAnyCallableOptionalUnion)version) TensorTypeis_torch_availableis_vision_availablelogging)ParameterFormat compute_effective_axis_dimension"compute_serialized_parameters_size)PretrainedConfigFeatureExtractionMixinImageProcessingMixinPreTrainedTokenizerBase)Image lcfeZdZUdZeed<eed<eed<dZe eed<dZ e eed<dS) PatchingSpeca Data class that holds patching specifications. Args: o: Module / object where the op to patch is located name: Name of the op to monkey patch custom_op: Custom op that patches the original op orig_op: Original op that is being patched op_wrapper: Wrapper (optional) that wraps both the original and custom ops. It is useful for ops that are class or static methods for instance. oname custom_opNorig_op op_wrapper) __name__ __module__ __qualname____doc__r __annotations__strr r%r r&l/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/transformers/onnx/config.pyr!r!0sf   FFF III"&GXh &&&%)J")))))r.r!cFeZdZdZdZdZdZejdZ e dddd ie d ddd ie dddd ie ddd ddd ddd d e dddd ie dddd ie dd diie ddd ddd d e ddd ddd de ddddddie dddd ie dd diie dddd ie dddd ie dddd idZ dIddde de eefdZedJddde ddfdZeedee eee fffdZedee eee fffdZede ee effd Zedefd!Zedefd"Zedefd#Zedefd$Zedefd%Z ede!fd&Z"e#d'ede!fd(Z$ dKd+ed,ed-ed.efd/Z% dLd+ed3ed4ed5efd6Z& dMd9e'd:d+ed;ede e(d,ed.ed-ed3ed4ed5ed?e d@dee effdAZ)dBee efdee effdCZ*dDZ+dEZ,edFe dGe-ede.e effdHZ/dS)N OnnxConfigzv Base class for ONNX exportable model describing metadata on how to export the model through the ONNX format. rz1.8logitsbatchsequencerrlast_hidden_state)r4 pred_boxes pred_masksr)r4r9) start_logits end_logits num_labelsheightwidth)rrrdecoder_sequence)z causal-lmdefaultzimage-classificationzimage-segmentationz masked-imz masked-lmmultiple-choicezobject-detectionzquestion-answeringzsemantic-segmentationz seq2seq-lmzsequence-classificationztoken-classificationz vision2seq-lmz speech2seq-lmrBNconfigrtaskpatching_specscL||_||jvr+t|d|j||_g|_||ngD]S}|}|j.tj|t|j |j }|j |TdS)Nz+ is not a supported task, supported tasks: )r%) _config_tasks_to_common_outputs ValueErrorkeysrE_patching_specsr% dataclassesreplacegetattrr"r#append)selfrDrErFspec final_specs r/__init__zOnnxConfig.__init__ps t4 4 4jjDDaDfDfDhDhjj  !&4&@NNb 4 4DJ|#(0wtvty?Y?YZZZ  ' ' 3 3 3 3  4 4r.returnc|||S)z Instantiate a OnnxConfig for a specific model Args: config: The model's configuration to use when exporting to ONNX Returns: OnnxConfig for this model )rEr-clsrDrEs r/from_model_configzOnnxConfig.from_model_configss6%%%%r.ct)z Mapping containing the axis definition of the input tensors to provide to the model Returns: For each input: its name associated to the axes symbolic name and the axis position within the tensor )NotImplementedErrorrQs r/inputszOnnxConfig.inputss"###r.cN|j|j}tj|S)z Mapping containing the axis definition of the output tensors to provide to the model Returns: For each output: its name associated to the axes symbolic name and the axis position within the tensor )rIrEcopydeepcopy)rQcommon_outputss r/outputszOnnxConfig.outputss#6tyA}^,,,r.c8t|jdrddiSdS)z Dictionary of keys to override in the model's config before exporting Returns: Dictionary with the keys (and their corresponding values) to override use_cacheFN)hasattrrHr\s r/values_overridezOnnxConfig.values_overrides( 4< - - (' 'tr.ctjS)zp The default batch size to use if no other indication Returns: Integer > 0 )r1default_fixed_batchr\s r/default_batch_sizezOnnxConfig.default_batch_sizes --r.ctjS)zu The default sequence length to use if no other indication Returns: Integer > 0 )r1default_fixed_sequencer\s r/default_sequence_lengthz"OnnxConfig.default_sequence_lengths 00r.ctjS)zw The default number of choices to use if no other indication Returns: Integer > 0 )r1default_fixed_num_choicesr\s r/default_num_choiceszOnnxConfig.default_num_choicess 33r.ctS)z{ Which onnx opset to use when exporting the model Returns: Integer ONNX Opset version )DEFAULT_ONNX_OPSETr\s r/default_onnx_opsetzOnnxConfig.default_onnx_opsets "!r.cdS)z What absolute tolerance value to use during model conversion validation. Returns: Float absolute tolerance value. gh㈵>r-r\s r/atol_for_validationzOnnxConfig.atol_for_validations tr.cxtr+ddlm}tj||jkSdS)z The minimum PyTorch version required to export the model. Returns: `bool`: Whether the installed version of PyTorch is compatible with the model. r)get_torch_versionF)rtransformers.utilsrvr parsetorch_onnx_minimum_version)rQrvs r/is_torch_support_availablez%OnnxConfig.is_torch_support_availablesK     < < < < < <=!2!2!4!4559XX X5r.num_parameterscHt|tjtkS)a Flag indicating if the model requires using external data format Args: num_parameters: Number of parameter on the model Returns: True if model.num_parameters() * size_of(float32) >= 2Gb False otherwise )rrFloatEXTERNAL_DATA_FORMAT_SIZE_LIMIT)r{s r/use_external_data_formatz#OnnxConfig.use_external_data_formats! /~?T U U. / r.r@( batch_size num_channels image_height image_widthcg}t|D]s}tj|||dz}|t j|ddt|S)Nuint8RGB) rangenprandomrandrPr fromarrayastypeconvert)rQrrrrimages_datas r/_generate_dummy_imagesz!OnnxConfig._generate_dummy_imagess|z"" P PA9>>, \JJSPD MM%/$++g*>*>??GGNN O O O O r."V@ sampling_rate time_duration frequencyc g}t|D]g}tjd|t||zd}|dtjdtjz|z|zzh|S)NrF)endpointg?r)rrlinspaceintrPsinpi)rQrrrr audio_datarts r/_generate_dummy_audioz OnnxConfig._generate_dummy_audios z"" G GA A}c--2O.P.P[`aaaA   cBF1ru9y+@1+D$E$EE F F F Fr.F preprocessor)rrr seq_length num_choicesis_pair framework tokenizerrcddlm}ddlm}ddlm}t ||r| td| 6tj dtt d| }t ||rPt|tjd }||}t|tj| }|jt'|jd kr|jnd }d |g|zg|z}|jd krttjd |z}|||}|D]3\}fdt1d t'D||<4t3||St3|||St ||r|jd dkr*td|jjd|jd t|tj}|||| |}t3|||St ||r^|jd dkrMt|tj}|||| |}t3|||St ||r^|jd dkrMt|tj}||| | | }t3|||Std)am Generate inputs to provide to the ONNX exporter for the specific framework Args: preprocessor: ([`PreTrainedTokenizerBase`], [`FeatureExtractionMixin`], or [`ImageProcessingMixin`]): The preprocessor associated with this model configuration. batch_size (`int`, *optional*, defaults to -1): The batch size to export the model for (-1 means dynamic axis). num_choices (`int`, *optional*, defaults to -1): The number of candidate answers provided for multiple choice task (-1 means dynamic axis). seq_length (`int`, *optional*, defaults to -1): The sequence length to export the model for (-1 means dynamic axis). is_pair (`bool`, *optional*, defaults to `False`): Indicate if the input is a pair (sentence 1, sentence 2) framework (`TensorType`, *optional*, defaults to `None`): The framework (PyTorch or TensorFlow) that the tokenizer will generate tensors for. num_channels (`int`, *optional*, defaults to 3): The number of channels of the generated images. image_width (`int`, *optional*, defaults to 40): The width of the generated images. image_height (`int`, *optional*, defaults to 40): The height of the generated images. sampling_rate (`int`, *optional* defaults to 22050) The sampling rate for audio data generation. time_duration (`float`, *optional* defaults to 5.0) Total seconds of sampling for audio data generation. frequency (`int`, *optional* defaults to 220) The desired natural frequency of generated audio. Returns: Mapping[str, Tensor] holding the kwargs to provide to the model's forward function rrrrNzPYou cannot provide both a tokenizer and a preprocessor to generate dummy inputs.ztThe `tokenizer` argument is deprecated and will be removed in version 5 of Transformers. Use `preprocessor` instead.zROverwriting the `preprocessor` argument with `tokenizer` to generate dummy inputs.r)fixed_dimensionnum_token_to_add0 rC) text_pairc*g|]}||zSr-r-).0irvs r/ z4OnnxConfig.generate_dummy_inputs..us')h)h)hQ!AK,?*@)h)h)hr.) tensor_type)return_tensors pixel_valuesz*The `preprocessor` is an image processor (zC) and expects `model_input_names[0]` to be "pixel_values", but got )r)rrinput_featuresz\Unable to generate dummy inputs for the model. Please provide a tokenizer or a preprocessor.) feature_extraction_utilsrimage_processing_utilsrtokenization_utils_baser isinstancerJwarningswarn FutureWarningloggerwarningrr1rhnum_special_tokens_to_addrk unk_tokenlenjoinrErnitemsrdictconvert_to_tensorsmodel_input_names __class__r'rr)rQrrrrrrrrrrrrrrrr token_to_add input_token dummy_inputtokenized_inputkrs ` @r/generate_dummy_inputsz OnnxConfig.generate_dummy_inputss` FEEEEEAAAAAAEEEEEE l$; < < qAVopp p  M+    NNo p p p$L l$; < <8 9J,J]^J(AA'JJL9J,M`lJ !*63|?U;V;VYZ;Z;Z&&  88[M22Z?@:MKy---?1Uhi *K7 ".,{k"R"R"R+1133iiDAq)h)h)h)h)h%PQSVWXSYSY[fJgJg)h)h)hOA&&O>>9>UUVVV [KKKLL L  &: ; ; -a0NBB qAWA`qqMYMklmMnqq :*V`VtuuuJ55j,P\^ijjK K RRRSS S  &< = = ,B`abBcguBuBu9*V`VtuuuJ55j,P\^ijjK K RRRSS S |%; < < AMA_`aAbfvAvAv:*V`VtuuuJ44ZP]_hiiK [KKKLL Ln r.reference_model_inputsc|S)a Generate inputs for ONNX Runtime using the reference model inputs. Override this to run inference with seq2seq models which have the encoder and decoder exported as separate ONNX files. Args: reference_model_inputs ([`Mapping[str, Tensor]`): Reference inputs for the model. Returns: `Mapping[str, Tensor]`: The mapping holding the kwargs to provide to the model's forward function r-)rQrs r/!generate_dummy_inputs_onnxruntimez,OnnxConfig.generate_dummy_inputs_onnxruntimes &%r.c|jD]E}|j|jn||j}t|j|j|FdSN)rLr&r$setattrr"r#)rQrRr$s r/ patch_opszOnnxConfig.patch_opssY( 2 2D*./*AtW[WeGfGfI DFDIy 1 1 1 1 2 2r.c|jD]E}|j|jn||j}t|j|j|FdSr)rLr&r%rr"r#)rQrRr%s r/ restore_opszOnnxConfig.restore_opssY( 0 0D&*o&=dll4??SWS_C`C`G DFDIw / / / / 0 0r.r#fieldclddlm}fdt||DS)a Flatten any potential nested structure expanding the name of the field with the index of the element within the structure. Args: name: The name of the nested structure field: The structure to, potentially, be flattened Returns: (dict[str, Any]): Outputs with flattened structure and key mapping this new structure. r)chainc&i|] \}}d||S).r-)ridxitemr#s r/ zAOnnxConfig.flatten_output_collection_property..s)]]])#t4#]]]r.) itertoolsr enumerate from_iterable)rXr#rrs ` r/"flatten_output_collection_propertyz-OnnxConfig.flatten_output_collection_propertysH $#####]]]]yATATUZA[A[7\7\]]]]r.)rBNrB)rr@rr)rrrr) rrrFNr@rrrrrN)0r'r(r)r*rhrkrnr rxryrrIr,r listr!rT classmethodrYpropertyrrrr]rbr rfrirlrorrfloatrtboolrz staticmethodrrrr rrrrrrrrr-r.r/r1r1Es !!.u!5!5 [(J,G,G!HII; 3Z5P5PQRR + X7z7R7R,S T T)k%*55")j99")j99    ![(J,G,G!HII [(J,G,G!HII&;1g,'?@@'K%*55")j99    *k$+ ; ;")j99    "-hG Yafm8n8n-o!p!p!k8=O-P-P"QRR#.;1g,/G#H#H + X7z7R7R,S T T$hG 0K0K%LMM$hG 0K0K%LMM?  Fqu44(4034QYZ^_kZlQm4444$ & &'9 & &Ua & & &[ &$WS#X%6 67$$$^X$-gc3h&7!78---X- '#s(*;!<   X .C...X.1111X14S444X4"C"""X"UX D   X      \ "fh14HK_bmp   25 NS fi     *.""9=vvghvv v  v  vJ'vvvvvvv56v c vvvvp &PSUXPXHY &^efiknfn^o & & & &222 000 ^c^(3-^TXY\^aYaTb^^^[^^^r.r1ceZdZ d dddedeeedeffd Ze d!ddded dfd Z e d e ee e effffd Ze d ee eeffd Ze d e fdZe d e fdZ d"ddde de dedeed e eeff fd Z d#de ee e effdedefdZdZdedeed eeefffd ZxZS)$OnnxConfigWithPastrBNFrDrrErFuse_pastc^t|||||_dS)N)rErF)superrTr)rQrDrErFrrs r/rTzOnnxConfigWithPast.__init__s/ d>JJJ  r.rUc|||dS)z Instantiate a OnnxConfig with `use_past` attribute set to True Args: config: The underlying model's config to use when exporting to ONNX Returns: OnnxConfig with `.use_past = True` T)rErr-rWs r/ with_pastzOnnxConfigWithPast.with_pastss6t4444r.cjtj}|jr||d|S)Nrb direction)rrbrfill_with_past_key_values_)rQrars r/rbzOnnxConfigWithPast.outputss8 = Q  + +Ni + P P Pr.cBt|jdr d|jiSdS)Nrd)rerHrr\s r/rfz"OnnxConfigWithPast.values_overrides( 4< - - 0/ /tr.cbt|jdstd|jjS)z The number of layers attribute retrieved from the model config. Override this for model configs where the number of layers attribute is not called `num_layers`. num_layerszcould not find the number of layers attribute in the model configuration, override the num_layers property of the model OnnxConfig to solve this)rerHAttributeErrorrr\s r/rzOnnxConfigWithPast.num_layerss= t|\22  B |&&r.cbt|jdstd|jjS)z The number of attention heads attribute retrieved from the model config. Override this for model configs where the number of attention heads attribute is not called `num_attention_heads`. num_attention_headszcould not find the number of attention heads attribute in the model configuration, override the num_attention_heads property of the model OnnxConfig to solve this)rerHrrr\s r/rz&OnnxConfigWithPast.num_attention_headss> t|%:;;  V |//r.rrrrrrrcXt|||||}|jrtst dddl}|dj\}} | dz} ||j| |jj |jzf} d|vrE|dj } | |d| || | gd |d<g|d <t|jD]E} |d || || fF|S) NrrrrACannot generate dummy past_keys inputs without PyTorch installed.r input_idsrattention_mask)dtyper)dimpast_key_values)rrrrrJtorchshaperrH hidden_sizercatonesrrrPzeros)rQrrrrr common_inputsrr5seqlenpast_key_values_lengthr mask_dtyperrs r/rz(OnnxConfigWithPast.generate_dummy_inputssg55 *W`i6   = b%''  !deee )+6C]3^3^ ///// 0$ e=DIc9d9d!T"5"5A"5"5"566=DIc9d9d!T"5"5A"5"5"566  e er.cJ|d||d|d<|d||d|d<dS)Nrrrrrr-rQflattened_outputr#rrs r/_flatten_past_key_values_z,OnnxConfigWithPast._flatten_past_key_values_BsH01!D,,3,,,-23A$D..3...///r.r#rci}|dvr.t|D]\}}|||||n"t||}|S)N)rr)rrrr)rQr#rrrrrs r/rz5OnnxConfigWithPast.flatten_output_collection_propertyFsy 1 1 1#E** O OQ../?sANNNN O %wwII$PUVV r.)rBNFrrrFN)F)r'r(r)r,r rr!rrTrrrrrrbr rfrrrrrrrrr __classcell__rs@r/rrs7; !!"!!!l!34 !  !!!!!! 5 51 5 5Ma 5 5 5[ 5gc3h&7!78X'#s(*;!<X  'C ' ' 'X ' 0S 0 0 0X 0*. ((,(( (  ( J' ( c ((((((Vqvee!(gc3h.?)?!@eMPeimeeee2777 s 8C= UYZ]_bZbUc          r.rc@eZdZedeeeeeffffd Zedeedfffd Z edeedfffd Z dd e d d ed ed e de e deeeff fd ZdeeeeeffdefdZdZxZS)OnnxSeq2SeqConfigWithPastrUctt|j}|D]6\}}d|vrdnd}|D]\}}d|vr|||<|||<7|jr||d|S)Nencoderencoder_sequencerAr6rbr)rrrbrrr)rQrar# axes_names sequence_nameaxis_idxrs r/rbz!OnnxSeq2SeqConfigWithPast.outputsRs1488@ . 4 4 6 6 0 0 D*2;t2C2C..I[M","2"2"4"4 0 0$%%+8Jx((,0Jx((  0 = Q  + +Ni + P P Pr..c tj}||f}nb#t$rUt|jdr.t|jdr|jj|jjf}ntdYnwxYw|S)Nencoder_layersdecoder_layerszcould not find the number of encoder and decoder layers attributes in the model configuration, override the num_layers property of the model OnnxConfig to solve this)rrrrerHr+r,)rQrrs r/rz$OnnxSeq2SeqConfigWithPast.num_layerscs +J$j1JJ   t|%566 74?? GDLZsDtDt '+|'KT\Mq&r##$$# #"r-rFNrrrrrrc tt||||||}|js|nd}tt||||||}d|D}t d i||} |jrt stdddl} | dj d} | dj d} | dj d}|j \} }| | | |j j | zf}| ||dz|j j |zf}g| d <|j \}}t||}t|||z }||krd nd }t!|D]m}| d | || || || |fn|d kr|n|}t!||D]E}| d | || |fF| S) Nrrc i|] \}}d|| S)decoder_r-)rr#tensors r/rzCOnnxSeq2SeqConfigWithPast.generate_dummy_inputs..s'___ f+T++V___r.rrrdecoder_input_idsr@rr%decoderr-)rrrrrrrrJrrrrHrrminmaxrrPr )rQrrrrrencoder_inputsdecoder_seq_lengthdecoder_inputsr rr5encoder_seq_lengthnum_encoder_attention_headsnum_decoder_attention_heads encoder_shape decoder_shapenum_encoder_layersnum_decoder_layersmin_num_layersmax_num_layersremaining_side_namerrrs r/rz/OnnxSeq2SeqConfigWithPast.generate_dummy_inputss1488NN *W`iO   04}CZZ!1488NN *9KU\hqO  `_H\H\H^H^___@@~@@@ =- b%''  !deee !+.4Q7E!.{!;!A!!D !./B!C!I!!L GKG_ D ')D+" (,GG M+"Q& (,GG M02M+ ,59_ 2  2 !35GHHN !35GHH>YN/ADV/V/V))\e >**  /077 M22 M22 M22 M22 &9I%E%EMM=E>>:: b b/077U9K9KU[[Y^M_M_8`aaaar.rrc |dvrtd|d|dkrdnd}|j\}}t||}t|||z }||krdnd}d } |dkrd nd } t |D]:} d | d ||d| d<d | d ||d| d<d | d ||d| d<d | d ||d| d<;t ||D]!} |dkrd | d } nd | d } | ||d| d|d<"dS)Nrrrr]rrr%r6past_encoder_sequencepast_decoder_sequencez past_decoder_sequence + sequencer5rr .decoder.key.decoder.value .encoder.key.encoder.valuer)rJrr7r8r) rQrrr#rArBrCrDrEr&rAr axes_infos r/rz4OnnxSeq2SeqConfigWithPast.fill_with_past_key_values_s 1 1 1iT]iiijj j$-$9$9  y26../1CDD/1CDD~U+=@R+R+RiiXa26?86K6K22Qs~&& _ _A?FK[;\;\ 77777 8AHM]=^=^ 99999 :?FK[;\;\ 77777 8AHM]=^=^ 99999 : :~~66 S SA"i// ',<== ',<== IR EEEE,?EEE F F  S Sr.c|d||d|d<|d||d|d<|d||d|d<|d||d|d <dS) NrrrIrrJrrKr@rLr-rs r/rz3OnnxSeq2SeqConfigWithPast._flatten_past_key_values_s89!D4434445:;A$D663666789!D4434445:;A$D663666777r.r)r'r(r)rrr,rrbtuplerrr rrr rrrr r!s@r/r#r#Qs gc3h&7!78X  E#s(O     X  #U38_ # # # # #X #$*. CC56CC C  C J' C c CCCCCCJSGCQTVYQYIZDZ<[ShkSSSS8???????r.r#)1r_rMrabcrr collectionsrcollections.abcrrtypingrr r r r numpyr packagingr utilsrrrrrrrconfiguration_utilsrrrrrrrPILr get_loggerr'rrqr~ dataclassr!r1rr#r-r.r/r[sr ##############--------@@@@@@@@@@@@@@PPPPPPPPPPPPhhhhhhhhhhB666666AAAAAA======AAAAAA  H % %#9 ********(v^v^v^v^v^v^v^v^r P P P P P SP P P fW?W?W?W?W? 2W?W?W?W?W?r.