Nofated095
/
so-vits-svc
mirror of https://github.com/svc-develop-team/so-vits-svc.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix#Corrected the issue of missing assignment for self.dtype and self.hps_ms under the only_diffusion path.

This commit is contained in:
Infko 2023-07-16 05:49:14 +00:00
parent cb4aeba1c1
commit caa4406646
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
inference/infer_tool.py
View File

@ -151,6 +151,15 @@ 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
if self.diffusion_args.train.amp_dtype == 'fp32':
self.dtype = torch.float32
elif self.diffusion_args.train.amp_dtype == 'fp16':
self.dtype = torch.float16
elif self.diffusion_args.train.amp_dtype == 'bf16':
self.dtype = torch.bfloat16
else:
raise ValueError(' [x] Unknown amp_dtype: ' + self.diffusion_args.train.amp_dtype)
self.hps_ms = utils.get_hparams_from_file(config_path, True)
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: