g%.ddlZddlZddlZddlmZddlmZddlmZddl m Z m Z m Z m Z ddlmZddlmZd d lmZmZd d lmZd d lmZd ZdgZdZdZdZedkr6ejdZe ej!edSdS)N)ArgumentParser)Path)version)AutoFeatureExtractorAutoImageProcessor AutoProcessor AutoTokenizer)logging)is_optimum_available)exportvalidate_model_outputs)FeaturesManager)get_preprocessorz1.5.0zvision-encoder-decoderctrSddlm}tj|}|tjt krt dt d|dnt dtjddd |j d |j |j d |j nd |j g}tj|tj }|t"ddS)Nr) __version__z&transformers.onnx requires optimum >= z but zT is installed. You can upgrade optimum by running: pip install -U optimum[exporters]zqtransformers.onnx requires optimum to run, you can install the library by running: pip install optimum[exporters]-mzoptimum.exporters.onnxz--model z--task z --framework )stdoutaThe export was done by optimum.exporters.onnx. We recommend using to use this package directly in future, as transformers.onnx is deprecated, and will be removed in v5. You can find more information here: https://huggingface.co/docs/optimum/exporters/onnx/usage_guides/export_a_model.)r optimum.versionrrparseMIN_OPTIMUM_VERSION RuntimeErrorsys executablemodelfeature frameworkoutput subprocessPopenPIPEwaitloggerinfo)argsoptimum_versionparsed_optimum_versioncmd_lineprocs V/var/www/html/ai-engine/env/lib/python3.11/site-packages/transformers/onnx/__main__.pyexport_with_optimumr-#s?  BBBBBB!(!?!? !GM2E$F$F F FT9LTTSbTTT  G  !     4: $,  +/>+E't~'''2 ;H  HZ_ = = =DIIKKK KK Zc N |jr|jn|jd|_|jjs |jjdt j|j|j |j |j }t j ||j\}}||j }|tvr|}|}||j }||j |j |j}|jt)|j|j|_|jt-|j|jkr6t/d|jd|dt-|j|jd t1j|j } t5| |||j|jjd \} } t7|| ||jjd | |jr|jn|jt=j|j } t5| |||j|jjd \} } t7|| ||jjd | |jr|jn|jt> d |jjd !d |jjd !dS|j"dkrtG|j } n|j"dkrt=j|j } n|j"dkrtIj|j } nb|j"dkrt1j|j } n=|j"dkrtKj|j } nt/d|j"d|j |j|_|j|jkr#t/d|jd|d|jd t5| |||j|j\} } |j |j|_t7|| ||j| |jt> d |j!tMj'dtPdS)Nz model.onnxT)parents)r cache_dir)rzOpset z is not sufficient to export z . At least z is required.zencoder_model.onnxzdecoder_model.onnxzAll good, model saved at: z, auto tokenizerimage_processorfeature_extractor processorzUnknown preprocessor type ''aThe export was done by transformers.onnx which is deprecated and will be removed in v5. We recommend using optimum.exporters.onnx in future. You can find more information here: https://huggingface.co/docs/optimum/exporters/onnx/usage_guides/export_a_model.))r is_filejoinpathparentexistsmkdirrget_model_from_featurerrrr1check_supported_model_or_raiseconfigENCODER_DECODER_MODELS get_encoder get_decoderget_encoder_configget_decoder_configopsetmaxdefault_onnx_opsetmin ValueErrorrfrom_pretrainedrratolatol_for_validationr r%r&as_posix preprocessorrrr warningswarn FutureWarning) r'r model_kindmodel_onnx_config onnx_config encoder_model decoder_modelencoder_onnx_configdecoder_onnx_configrN onnx_inputs onnx_outputss r,export_with_transformersr[Esp!%!4!4!6!6^$++DK@S@fggwww  ,;DJGG $*    J K  ' '(< = = % % ! \     K  ' '(< = =  ODII(;(O    %4TZ@@ $*    J K  ' '(< = = % % ! \     K  ' '(< = =  ODII(;(O       O);)D)DEY)Z)Z)c)c)e)e O O "++,@AAJJLL O O        & &+DJ77LL  + - -(8DDLL  "3 3 3-=djIILL  "5 5 5/? KKLL  + - -(8DDLLO4;LOOOPP P : $7DJ : 6 6 6KKK*KK(;KKK  %+    J K % % ! \ 9 #7DI{L%l\`\efff I1E1E1G1GIIJJJ  _       r.ctd}|ddtdd|ddd |d tdd |dtdd |dtddgdd|dt d|dtdd |dtgddd|ddd !|}|jstst|dSt|dS)"Nz'Hugging Face Transformers ONNX exporterrz--modelTz>Model ID on huggingface.co or path on disk to load model from.)typerequiredhelpz --featuredefaultz.The type of features to export the model with.)r`r_z--opsetz,ONNX opset version to export the model with.)r]r`r_z--atolz8Absolute difference tolerance when validating the model.z --frameworkpttfzThe framework to use for the ONNX export. If not provided, will attempt to use the local checkpoint's original framework or what is available in the environment.)r]choicesr`r_r z4Path indicating where to store generated ONNX model.)r]r_z --cache_dirz%Path indicating where to store cache.z--preprocessor)r2r3r5r4r6r2zKWhich type of preprocessor to use. 'auto' tries to automatically detect it.z--export_with_transformers store_truezWhether to use transformers.onnx instead of optimum.exporters.onnx to perform the ONNX export. It can be useful when exporting a model supported in transformers but not in optimum, otherwise it is not recommended.)actionr_) r add_argumentstrintfloatr parse_argsr[r r-)parserr's r,mainrls E F FF  icD7w  =   T@nooo ud1k  t  8     t2hiii  CDklll  ZZZ Z  $       D $",@,B,B" &&&&&D!!!!!r.__main__ztransformers.onnx)"r!rrOargparserpathlibr packagingrrrrr r utilsr utils.import_utilsr convertrrfeaturesrrrr@r-r[rl__name__ get_loggerr%setLevelINFOr.r,rzsn ######UUUUUUUUUUUU55555533333333%%%%%%######23Dt t t n0"0"0"f z W  3 4 4F OOGL!!!DFFFFFr.