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

Update infer_tool.py

This commit is contained in:
YuriHead 2023-07-16 22:13:51 +08:00 committed by GitHub
parent caa4406646
commit 59d7b10cb3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
inference/infer_tool.py
View File

@ -151,15 +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
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.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: