o
    4ѹgZ                     @   sj   d dl Z d dlZd dlZd dlmZmZmZ d dlmZ ej	ddddefddZ
edkr3e
  dS dS )	    N)
DictConfig	OmegaConf
ListConfig)	AutoModel)Zconfig_nameZversion_basecfgc              
      s    fdd  | }| ddrdd l}|  d|vr d|d< tdi |}|j| dd | d	d
| dd| dd| dd| ddd}t| d S )Nc                    s<   t | trtj| ddS t | tr fdd|  D S | S )NT)resolvec                    s   i | ]	\}}| |qS  r   ).0kvto_plain_listr   U/Users/admin/.pyenv/versions/3.10.0/lib/python3.10/site-packages/funasr/bin/export.py
<dictcomp>   s    z5main_hydra.<locals>.to_plain_list.<locals>.<dictcomp>)
isinstancer   r   Zto_containerr   items)Zcfg_itemr   r   r   r      s
   

z!main_hydra.<locals>.to_plain_listdebugFr   ZdevicecpuinputtypeZonnxquantizezfallback-num   	calib_numd   opset_version   )r   r   r   Zfallback_numr   r   r   )getpdb	set_tracer   Zexportprint)r   kwargsr   modelresr   r   r   
main_hydra	   s"   





r#   __main__)osZhydraloggingZ	omegaconfr   r   r   Zfunasr.auto.auto_modelr   mainr#   __name__r   r   r   r   <module>   s    
