
    fPi                         d dl Z d dlZd Zdej        defdZedk    rB e            Z ej        ej	                  Z
 ee
            ej        e
ej                   dS dS )    Nc                      t          j                    } |                     ddd           |                     ddd           |                                 }|S )Nz--inputTzinput model)requiredhelpz--outputzoutput model)argparseArgumentParseradd_argument
parse_args)parserargss     /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/onnxruntime/tools/remove_initializer_from_input.pyget_argsr      s\    $&&F
	D}EEE

TGGGDK    modelreturnc                     | j         dk     rt          d           dS | j        j        }i }|D ]}|||j        <   d}| j        j        D ]-}|j        |v r"d}|                    ||j                            .|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    !\]]]u[FM * *$)ej!!H{. ; ;},,HMM-(89:::Or   __main__)r   onnxr   
ModelProtoboolr   __name__r   loadr   r   saveoutput r   r   <module>r'      s        T    & z8::DDIdj!!E!!%(((DIeT[!!!!!	 r   