g[ddlZddlZddlZddlZddlmZmZmZmZejddejjddjejejdZd Zd Zed k(rXeZej(d kred ej(dej,j/ej0Zesej4ej0de_ej9dej:sde_ej<s#ej9dej>eej@s#ej9dejBeejDs#ej9dejFeejHs$ej9dejJeyyy)N)chain_enc_dec_with_beamsearchexport_summarization_edinit!export_summarization_enc_dec_pastonnx_inferencezO%(asctime)s | %(levelname)s | %(name)s | [%(filename)s:%(lineno)d] %(message)sz%Y-%m-%d %H:%M:%SLOGLEVELINFO)formatdatefmtlevelstreamgeneratec lt|D]&}tj|dt||(y)Nz: )varsloggerinfogetattr)argsargs `/var/www/openai/venv/lib/python3.12/site-packages/onnxruntime/transformers/models/bart/export.py print_argsrs/Dz se2gdC0123ctjd}|jdtdd|jdtdd |jd d td d |jddtdd|jddtdd|jddtd|jddtdd|jdt dd|jdtd d!|jd"t dd#|jd$td%d&|jd'd()|jd*d()|jd+d()|jd,d()|jd-}|jd.d/td0d12t|j|jS)3NF)add_helpz --max_lengthz default to 20)typedefaulthelpz --min_lengthrz default to 0z-oz--output 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_beamsz default to 5z--repetition_penaltyg?zdefault to 1.0z--no_repeat_ngram_sizez default to 3z--early_stoppingzdefault to Falsez--opset_versionz minimum is 14z --no_encoder store_true)actionz --no_decoderz --no_chainz--no_inferencezrequired input argumentsz-mz --model_dirTzThe directory contains input huggingface model. An official model like facebook/bart-base is also acceptable.)rrequiredr) argparseArgumentParser add_argumentintstrfloatbooladd_argument_groupr parse_args) parent_parser required_argss r user_commandr0"s++U;M~C/Z~CXtZc=Wuvt^#tR^_t\TPstt^#DVWt]an]5E3Uef7c1Sab1eRde0sB_]~lC~lC|LA/ E!445OPM  ^ }'')*  # # %%r__main__r!z=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%loggingossysutilsrrrr basicConfigenvirongetupperstdout getLoggerrrr0__name__r opset_version ValueErrorpathexistsoutputisExistmakedirsdevicer input_text no_encoderexport_encoder no_decoderexport_decoderno_chain convert_model no_inference run_inferencerrrQs  \  **..V , 2 2 4 ::    : &4 &B z >D BXY]YkYkXllmnooggnnT[[)G  DKK DK KK ?? u  ?? =>2#2248 ?? =>8)88> == =>3%33D9    GH$$$T* ;r