U T?h[ã@sLddlZddlZddlZddlZddlmZmZmZmZej ddej   dd¡  ¡ej de d¡Zd d „Zd d „Zed krHeƒZejdkr edej›dƒ‚ej ej¡Zes¾e ej¡de_e d¡ejsÚde_ejsôe d¡e e¡ej se d¡e !e¡ej"s,e d¡e #e¡ej$sHe d¡e %e¡dS)éN)Úchain_enc_dec_with_beamsearchÚexport_summarization_edinitÚ!export_summarization_enc_dec_pastÚonnx_inferencezO%(asctime)s | %(levelname)s | %(name)s | [%(filename)s:%(lineno)d] %(message)sz%Y-%m-%d %H:%M:%SZLOGLEVELÚINFO)ÚformatÚdatefmtÚlevelÚstreamÚgeneratecCs,t|ƒD]}t |›dt||ƒ›¡qdS)Nz: )ÚvarsÚloggerÚinfoÚgetattr)ÚargsÚarg©rú]/var/www/html/venv/lib/python3.8/site-packages/onnxruntime/transformers/models/bart/export.pyÚ print_argss rcCsDtjdd}|jdtddd|jdtdd d|jd d td d d|jddtddd|jddtddd|jddtdd|jddtddd|jdtddd|jdtd d!d|jd"tdd#d|jd$td%d&d|jd'd(d)|jd*d(d)|jd+d(d)|jd,d(d)| d-¡}|jd.d/td0d1d2t|  ¡ƒ|  ¡S)3NF)Úadd_helpz --max_lengthéz default to 20)ÚtypeÚdefaultÚhelpz --min_lengthrz default to 0z-oz--outputZ onnx_modelszdefault name is onnx_models.z-iz --input_textz input textz-sz --spm_pathz!tokenizer model from sentencepicez-vz --vocab_pathzvocab dictionary)rrz-bz --num_beamséz default to 5z--repetition_penaltygð?zdefault to 1.0z--no_repeat_ngram_sizeéz default to 3z--early_stoppingzdefault to Falsez--opset_versionéz minimum is 14z --no_encoderÚ store_true)Úactionz --no_decoderz --no_chainz--no_inferencezrequired input argumentsz-mz --model_dirTzŒThe directory contains input huggingface model. An official model like facebook/bart-base is also acceptable.)rÚrequiredr) ÚargparseÚArgumentParserÚ add_argumentÚintÚstrÚfloatÚboolÚadd_argument_grouprÚ parse_args)Z parent_parserZ required_argsrrrÚ user_command"s4  û r)Ú__main__rz=The minimum supported opset version is 14! The given one was Ú.Úcpuz ENV: CPU ...aPG&E stated it scheduled the blackouts in response to forecasts for high winds amid dry conditions. The aim is to reduce the risk of wildfires. Nearly 800 thousand customers were scheduled to be affected by the shutoffs which were expected to last through at least midday tomorrow.z'========== EXPORTING ENCODER ==========z'========== EXPORTING DECODER ==========z'========== CONVERTING MODELS ==========z1========== INFERENCING WITH ONNX MODEL ==========)&r ÚloggingÚosÚsysÚutilsrrrrÚ basicConfigÚenvironÚgetÚupperÚstdoutÚ getLoggerr rr)Ú__name__rZ opset_versionÚ ValueErrorÚpathÚexistsÚoutputZisExistÚmakedirsZdevicerZ input_textZ no_encoderZexport_encoderZ no_decoderZexport_decoderZno_chainZ convert_modelZ no_inferenceZ run_inferencerrrrÚsJü !    ÿ