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()