o
    ߥi                     @   sx   d dl Z d dlmZmZ d dlmZ d dlmZ d dlm	Z	 d dl
mZmZ dgZe	jejejdG d	d deZdS )
    N)AnyDict)Models)Model)MODELS)
FrameworksTasksLauraCodecGenModel)module_namec                       sH   e Zd Zdededeeef f fddZdeeef fddZ  ZS )	r	   	model_dir
model_namemodel_configc                    s<   t  j|||g|R i | ||tj|||d| _dS )a  initialize the info of model.

        Args:
            model_dir (str): the model path.
            model_name (str): the itn model name from configuration.json
            model_config (Dict[str, Any]): the detail config about model from configuration.json
        )model_workspacer   
model_pathr   N)super__init__ospathjoin	model_cfg)selfr   r   r   argskwargs	__class__ [/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/audio/tts/laura_codec.pyr      s   	zLauraCodecGenModel.__init__returnc                 C   s   | j S )z1
          just return the model config

        )r   )r   r   r   r   forward&   s   zLauraCodecGenModel.forward)	__name__
__module____qualname__strr   r   r   r   __classcell__r   r   r   r   r	      s
    

)r   typingr   r   modelscope.metainfor   modelscope.models.baser   modelscope.models.builderr   modelscope.utils.constantr   r   __all__register_moduletext_to_speechlaura_codecr	   r   r   r   r   <module>   s   