o
    )j                     @   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
        )Zmodel_workspacer   Z
model_pathr   N)super__init__ospathjoin	model_cfg)selfr   r   r   argskwargs	__class__ h/var/www/html/Deteccion_Ine/venv/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   Zmodelscope.metainfor   Zmodelscope.models.baser   Zmodelscope.models.builderr   Zmodelscope.utils.constantr   r   __all__Zregister_moduleZtext_to_speechZlaura_codecr	   r   r   r   r   <module>   s   