o
    iZ                     @   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)config_name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
   E/home/ubuntu/.local/lib/python3.10/site-packages/funasr/bin/export.py
<dictcomp>   s    z5main_hydra.<locals>.to_plain_list.<locals>.<dictcomp>)
isinstancer   r   to_containerr   items)cfg_itemr   r
   r   r      s
   

z!main_hydra.<locals>.to_plain_listdebugFr   devicecpuinputtypeonnxquantizezfallback-num   	calib_numd   opset_version   )r   r   r   fallback_numr   r    r
   )getpdb	set_tracer   exportprint)r   kwargsr$   modelresr
   r   r   
main_hydra	   s"   





r+   __main__)oshydralogging	omegaconfr   r   r   funasr.auto.auto_modelr   mainr+   __name__r
   r
   r
   r   <module>   s    
