
    g                         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              yy)    Nreturnc                  
   t        j                         } | j                  dt        d       | j                  dt        d       | j                  dt        d       | j                  dt        d       | j                         S )Nz--npz_file_pathT)typerequiredz--output_file_pathz--adapter_versionz--model_version)argparseArgumentParseradd_argumentstrint
parse_args)parsers    a/var/www/openai/venv/lib/python3.12/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                    t        j                         }|j                  |       |j                  |       i }t	        j
                  |       5 }|j                         D ])  \  }}t         j                  j                  |      }	|	||<   + 	 ddd       |j                  |       |j                  |       y# 1 sw Y   ,xY w)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   =B77C c                      t               } t        | j                  | j                  | j                  | j
                         y)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      s|     	 
  ( 4;;4144EH4\^\g\g4"c  zCHHTV r   