
    pi                         d dl Z d dlZd Zdej                  defdZedk(  rI e       Z ej                  ej                        Z
 ee
        ej                  e
ej                         yy)    Nc                      t        j                         } | j                  ddd       | j                  ddd       | j                         }|S )Nz--inputTzinput model)requiredhelpz--outputzoutput model)argparseArgumentParseradd_argument
parse_args)parserargss     q/opt/services/ai/voice_agent/venv/lib/python3.12/site-packages/onnxruntime/tools/remove_initializer_from_input.pyget_argsr      sM    $$&F
	D}E

TGDK    modelreturnc                 0   | j                   dk  rt        d       y| j                  j                  }i }|D ]  }|||j                  <    d}| j                  j
                  D ]1  }|j                  |v sd}|j                  ||j                            3 |S )N   zLModel with ir_version below 4 requires to include initializer in graph inputFT)
ir_versionprintgraphinputnameinitializerremove)r   inputsname_to_inputr   modifiedr   s         r   remove_initializer_from_inputr      s    !\][[FM *$)ejj!* H{{.. ;},HMM-(8(89:;
 Or   __main__)r   onnxr   
ModelProtoboolr   __name__r   loadr   r   saveoutput r   r   <module>r'      sm      T & z:DDIIdjj!E!%(DIIeT[[!	 r   