From 8c255811adc8c184f85d0572760d2770c1ad7ecb Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 25 Jan 2022 17:57:16 -0700 Subject: [PATCH] more fixes --- codes/data/audio/fast_paired_dataset.py | 2 +- codes/models/gpt_voice/unet_diffusion_tts_experimental.py | 2 +- codes/train.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/codes/data/audio/fast_paired_dataset.py b/codes/data/audio/fast_paired_dataset.py index e1be1bbb..638ec81a 100644 --- a/codes/data/audio/fast_paired_dataset.py +++ b/codes/data/audio/fast_paired_dataset.py @@ -54,9 +54,9 @@ class FastPairedVoiceDataset(torch.utils.data.Dataset): self.conditioning_candidates = opt_get(hparams, ['num_conditioning_candidates'], 1) self.conditioning_length = opt_get(hparams, ['conditioning_length'], 44100) self.debug_failures = opt_get(hparams, ['debug_loading_failures'], False) - self.aligned_codes_to_audio_ratio = opt_get(hparams, ['aligned_codes_ratio'], 443) self.text_cleaners = hparams.text_cleaners self.sample_rate = hparams.sample_rate + self.aligned_codes_to_audio_ratio = 443 * self.sample_rate // 22050 self.max_wav_len = opt_get(hparams, ['max_wav_length'], None) if self.max_wav_len is not None: self.max_aligned_codes = self.max_wav_len // self.aligned_codes_to_audio_ratio diff --git a/codes/models/gpt_voice/unet_diffusion_tts_experimental.py b/codes/models/gpt_voice/unet_diffusion_tts_experimental.py index 6e2cd9d8..1074b706 100644 --- a/codes/models/gpt_voice/unet_diffusion_tts_experimental.py +++ b/codes/models/gpt_voice/unet_diffusion_tts_experimental.py @@ -334,7 +334,7 @@ class DiffusionTts(nn.Module): emb = emb1 # Mask out guidance tokens for un-guided diffusion. - if self.nil_guidance_fwd_proportion > 0: + if self.training and self.nil_guidance_fwd_proportion > 0: token_mask = torch.rand(tokens.shape, device=tokens.device) < self.nil_guidance_fwd_proportion tokens = torch.where(token_mask, self.mask_token_id, tokens) diff --git a/codes/train.py b/codes/train.py index 812b4b56..9a2e8a80 100644 --- a/codes/train.py +++ b/codes/train.py @@ -299,7 +299,7 @@ class Trainer: if __name__ == '__main__': parser = argparse.ArgumentParser() - parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_diffusion_tts_medium.yml') + parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../experiments/train_diffusion_tts_experimental_fp16/train_diffusion_tts.yml') parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none', help='job launcher') parser.add_argument('--local_rank', type=int, default=0) args = parser.parse_args()