
    h                         S SK r S SKrS SKrS SKrS SKrS\ 4S jrS\R                  S\	S\	S\R                  4S jr
S\	4S	 jr\S
:X  a  \R                  " \" 5       5        gg)    Nreturnc                      [         R                  " 5       n U R                  S[        SS9  U R                  S[        SS9  U R                  S[        SS9  U R                  S[        SS9  U R                  5       $ )Nz--npz_file_pathT)typerequiredz--output_file_pathz--adapter_versionz--model_version)argparseArgumentParseradd_argumentstrint
parse_args)parsers    f/var/www/fran/franai/venv/lib/python3.13/site-packages/onnxruntime/capi/convert_npz_to_onnx_adapter.pyget_argsr      sw    $$&F
)dC
,3F
+#E
)dC    npz_file_pathadapter_versionmodel_versionoutput_file_pathc                    [         R                  " 5       nUR                  U5        UR                  U5        0 n[        R
                  " U 5       nUR                  5        H(  u  px[         R                  R                  U5      n	XU'   M*     SSS5        UR                  U5        UR                  U5        g! , (       d  f       N1= f)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$$]3		4 JJLLD88@I&/d# ) 
 
 !!"34!!"23 
 	s   =B88
Cc                      [        5       n [        U R                  U R                  U R                  U R
                  5        g)Nr   )r   r'   r   r   r   r   )argss    r   mainr*   )   s3    :D4--t/C/CTEWEWY]YnYnor   __main__)r   ossysnumpyr   onnxruntimer   r   PathLiker   r'   r*   __name__exit r   r   <module>r4      sz     	 
  ( 4;;4144EH4\^\g\g4"c  zHHTV r   