Merge pull request #307 from hxdnshx/finn/only_diffusion_fix

Fixed the error occurring under the only_diffusion path
2023-07-16 22:14:52 +08:00 · 2023-07-16 22:14:52 +08:00 · 85c6388070
parent 847e71c3d7 59d7b10cb3
commit 85c6388070
1 changed files with 1 additions and 0 deletions
--- a/inference/infer_tool.py
+++ b/inference/infer_tool.py
@ -151,6 +151,7 @@ class Svc(object):
                    self.target_sample = self.diffusion_args.data.sampling_rate
                    self.hop_size = self.diffusion_args.data.block_size
                    self.spk2id = self.diffusion_args.spk
+                    self.dtype = torch.float32
                    self.speech_encoder = self.diffusion_args.data.encoder
                    self.unit_interpolate_mode = self.diffusion_args.data.unit_interpolate_mode if self.diffusion_args.data.unit_interpolate_mode is not None else 'left'
                if spk_mix_enable: