Debug unit2mel.py

Debug unit2mel.py
This commit is contained in:
YuriHead 2023-06-23 21:16:39 +08:00 committed by GitHub
parent 4f43531207
commit cf35e5f77f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -49,7 +49,7 @@ def load_model_vocoder(
model.to(device)
model.load_state_dict(ckpt['model'])
model.eval()
print(f'Loaded diffusion model, sampler is {ckpt["infer"]["methold"]}, speedup: {ckpt["infer"]["speedup"]} ')
print(f'Loaded diffusion model, sampler is {args.infer.method}, speedup: {args.infer.speedup} ')
return model, vocoder, args