This commit is contained in:
Johan Nordberg 2022-06-11 21:00:16 +09:00
parent 3791eb7267
commit dba14650cb
2 changed files with 2 additions and 2 deletions

View File

@ -210,7 +210,7 @@ tts = TextToSpeech(models_dir=args.models_dir, enable_redaction=not args.disable
device=args.device, autoregressive_batch_size=args.batch_size) device=args.device, autoregressive_batch_size=args.batch_size)
gen_settings = { gen_settings = {
'use_deterministic_seed': seed, 'use_deterministic_seed': seed,
'varbose': not args.quiet, 'verbose': not args.quiet,
'k': args.candidates, 'k': args.candidates,
'preset': args.preset, 'preset': args.preset,
} }

View File

@ -270,7 +270,7 @@ class TextToSpeech:
if not isinstance(voice_samples, list): if not isinstance(voice_samples, list):
voice_samples = [voice_samples] voice_samples = [voice_samples]
for vs in voice_samples: for vs in voice_samples:
auto_conds.append(format_conditioning(vs, self.device)) auto_conds.append(format_conditioning(vs, device=self.device))
auto_conds = torch.stack(auto_conds, dim=1) auto_conds = torch.stack(auto_conds, dim=1)
self.autoregressive = self.autoregressive.to(self.device) self.autoregressive = self.autoregressive.to(self.device)
auto_latent = self.autoregressive.get_conditioning(auto_conds) auto_latent = self.autoregressive.get_conditioning(auto_conds)