
    h                         S SK r S SKrS rS\R                  S\4S jr\S:X  aI  \" 5       r\R                  " \R                  5      r
\" \
5        \R                  " \
\R                  5        gg)    Nc                      [         R                  " 5       n U R                  SSSS9  U R                  SSSS9  U R                  5       nU$ )Nz--inputTzinput model)requiredhelpz--outputzoutput model)argparseArgumentParseradd_argument
parse_args)parserargss     i/var/www/fran/franai/venv/lib/python3.13/site-packages/onnxruntime/tools/remove_initializer_from_input.pyget_argsr      sM    $$&F
	D}E

TGDK    modelreturnc                 <   U R                   S:  a  [        S5        gU R                  R                  n0 nU H  nX2UR                  '   M     SnU R                  R
                   H4  nUR                  U;   d  M  SnUR                  X%R                     5        M6     U$ )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'      si      T & z:DIIdjj!E!%(IIeT[[!	 r   