o @Ti @sddlZddlZddlmZedZeejeZ e eje e de e dej fddZdeejdBfdd Z dd ed edBd edBdeejeffd dZddededejfddZddZdejdejfddZdS)Nkanade_tokenizerz1[%(asctime)s] %(levelname)s %(name)s: %(message)sreturncCstS)N)loggerrrB/home/ubuntu/.local/lib/python3.10/site-packages/linacodec/util.py get_loggersrmodulescCs*|D]}|dur|D]}d|_q qdS)NF) parameters requires_grad)rmoduleparamrrrfreeze_moduless  r path frame_offset num_framescCsxddl}||'}|dur|||j|pdddd}t|j}|j}Wd||fS1s3wY||fS)Nrfloat32T)framesdtype always_2d) soundfile SoundFileseekreadtorch from_numpyT samplerate)rrrsffrwaveform sample_raterrr_load_audio_internals    r"] audio_pathr!cCsxddl} t|\}}|jddkrtj|ddd}||kr)|j||}||}tt|d}||}| dS)NrT)dimkeepdimg:0yE>) torchaudior"shapermean transformsResamplemaxabssqueeze)r$r!r(r sr resamplermax_valrrr load_audio,s  r3cCs"ddlm}|d}|}|S)Nr)Vocoszcharactr/vocos-mel-24khz)vocosr4from_pretrainedeval)r4modelrrr load_vocoderBs  r9mel_spectrogramcCs|tj}||}|S)a(Convert mel spectrogram to waveform using Vocos vocoder. Args: vocoder (Vocos): Pretrained Vocos vocoder. mel_spectrogram (torch.Tensor): Input mel spectrogram tensor (..., n_mels, frame). Returns: torch.Tensor: Generated audio waveform tensor (..., samples). )torrdecode)vocoderr:generated_waveformrrrvocodeJs  r?)NN)r#)loggingrtorch.nnnn getLoggerrsetLevelINFO StreamHandlerhandler setFormatter Formatter addHandlerLoggerrlistModuler strinttupleTensorr"r3r9r?rrrrs0