(self)
| 14 | |
| 15 | class DiffFsTask(FastSpeech2Task): |
| 16 | def build_tts_model(self): |
| 17 | mel_bins = hparams['audio_num_mel_bins'] |
| 18 | self.model = GaussianDiffusion( |
| 19 | phone_encoder=self.phone_encoder, |
| 20 | out_dims=mel_bins, denoise_fn=DIFF_DECODERS[hparams['diff_decoder_type']](hparams), |
| 21 | timesteps=hparams['timesteps'], |
| 22 | loss_type=hparams['diff_loss_type'], |
| 23 | spec_min=hparams['spec_min'], spec_max=hparams['spec_max'], |
| 24 | ) |
| 25 | |
| 26 | def run_model(self, model, sample, return_output=False, infer=False): |
| 27 | txt_tokens = sample['txt_tokens'] # [B, T_t] |
nothing calls this directly
no test coverage detected