
    fPi                         d dl Z d dlZd dlZd dlZd dlZde fdZdej        de	de	dej        fdZ
de	fd	Zed
k    r ej         e                       dS dS )    Nreturnc                  8   t          j                    } |                     dt          d           |                     dt          d           |                     dt          d           |                     dt          d           |                                 S )Nz--npz_file_pathT)typerequiredz--output_file_pathz--adapter_versionz--model_version)argparseArgumentParseradd_argumentstrint
parse_args)parsers    /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/onnxruntime/capi/convert_npz_to_onnx_adapter.pyget_argsr      s    $&&F
)dCCC
,3FFF
+#EEE
)dCCC    npz_file_pathadapter_versionmodel_versionoutput_file_pathc                    t          j                    }|                    |           |                    |           i }t	          j        |           5 }|                                D ])\  }}t           j                            |          }	|	||<   *	 ddd           n# 1 swxY w Y   |	                    |           |
                    |           dS )zCThe function converts lora parameters in npz to onnx_adapter formatN)ortAdapterFormatset_adapter_versionset_model_versionnploaditemsOrtValueortvalue_from_numpyset_parametersexport_adapter)
r   r   r   r   adapter_formatname_to_ort_valuedatanamenp_arr	ort_values
             r   export_lora_parametersr'      s    &((N&&777$$]333			 04 JJLL 	0 	0LD&88@@I&/d##	00 0 0 0 0 0 0 0 0 0 0 0 0 0 0
 !!"3444!!"233333s   ?B  B$'B$c                  n    t                      } t          | j        | j        | j        | j                   dS )Nr   )r   r'   r   r   r   r   )argss    r   mainr*   )   s1    ::D4-t/CTEWY]Ynooo1r   __main__)r   ossysnumpyr   onnxruntimer   r   PathLiker   r'   r*   __name__exit r   r   <module>r4      s     				 



        (    4;4144EH4\^\g4 4 4 4"c     zCHTTVV r   