o
    
ja
                     @   s   d dl Z d dlmZmZ d dlmZmZ d dlmZm	Z	 d dl
mZ d dlmZmZ d dlmZ eG dd	 d	eZd
d ZedkrGe  dS dS )    N)	dataclassfield)TrainerTrainerArgs)load_configregister_config)AudioProcessor)load_wav_dataload_wav_feat_data)setup_modelc                   @   s&   e Zd ZU edddidZeed< dS )TrainVocoderArgsNhelpzPath to the config file.)defaultmetadataconfig_path)__name__
__module____qualname__r   r   str__annotations__ r   r   H/home/kuhnn/.local/lib/python3.10/site-packages/TTS/bin/train_vocoder.pyr      s   
 r   c               
   C   sH  t  } | jdd}| \}}| | |js|jr_|jr0t|j}t|dkr/|j|dd n/|jrKttj	
|jd}t|dkrJ|j|dd nddlm} | }|| t|j }d|v rz|jrztd	|j  t|j|j|j\}}n	t|j|j\}}tdi |j}	t|}
t| ||j|
||d
|	idd}|  dS )z:Run `tts` model training directly by a `config.json` file. )
arg_prefixr   T)relaxed_parserzconfig.json)BaseTrainingConfigfeature_pathz > Loading features from: audio_processorF)modeltrain_sampleseval_samplestraining_assetsparse_command_line_argsNr   )r   init_argparseparse_known_args
parse_argsr   continue_pathr   lenospathjoinTTS.config.shared_configsr   r   r   r   printr
   	data_patheval_split_sizer	   r   audior   r   output_pathfit)
train_argsparserargsconfig_overridesconfigr   config_baser    r   apr   trainerr   r   r   main   sH   



r:   __main__)r(   dataclassesr   r   r9   r   r   
TTS.configr   r   TTS.utils.audior   TTS.vocoder.datasets.preprocessr	   r
   TTS.vocoder.modelsr   r   r:   r   r   r   r   r   <module>   s    ;
