o
    3Ih                     @   sB   d dl Z d dlZd dlZddlmZ dd Zedkre  dS dS )    N   )update_onnx_opsetc                  C   sx   t jtjt dtj dd} | jdt	dddd | jd	t
jd
d | jdt
jdd |  }t|j|j|j d S )N:a  
                                     Update the ONNX opset of the model.
                                     New opset must be later than the existing one.
                                     If not specified will update to opset 15.
                                     )descriptionz--opsetF   zONNX opset to update to.)typerequireddefaulthelpinput_modelz%Provide path to ONNX model to update.)r   r
   output_modelz,Provide path to write updated ONNX model to.)argparseArgumentParserospathbasename__file__update_onnx_opset_helper__name__add_argumentintpathlibPath
parse_argsr   r   opsetr   )parserargs r   c/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/onnxruntime/tools/update_onnx_opset.pyr      s   	r   __main__)r   r   r   onnx_model_utilsr   r   r   r   r   r   r   <module>   s   
