o )wi@sdddlZddlZddZdejdefddZedkr0eZeej Z ee e e ej dSdS) NcCs4t}|jdddd|jdddd|}|S)Nz--inputTz input model)requiredhelpz--outputz output model)argparseArgumentParser add_argument parse_args)parserargsr l/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnxruntime/tools/remove_initializer_from_input.pyget_argss r modelreturncCsj|jdkr tddS|jj}i}|D]}|||j<qd}|jjD]}|j|vr2d}|||jq!|S)NzLModel with ir_version below 4 requires to include initializer in graph inputFT) ir_versionprintgraphinputname initializerremove)r inputs name_to_inputrmodifiedrr r r remove_initializer_from_inputs    r__main__) ronnxr ModelProtoboolr__name__r loadrr saveoutputr r r r s