From e6824e398f96359f970e594a39a458d2ca708940 Mon Sep 17 00:00:00 2001 From: James Betker Date: Wed, 23 Feb 2022 21:21:45 -0700 Subject: [PATCH] Load dvae to cpu --- codes/scripts/audio/gen/speech_synthesis_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/scripts/audio/gen/speech_synthesis_utils.py b/codes/scripts/audio/gen/speech_synthesis_utils.py index a2c902b0..62fee92c 100644 --- a/codes/scripts/audio/gen/speech_synthesis_utils.py +++ b/codes/scripts/audio/gen/speech_synthesis_utils.py @@ -14,7 +14,7 @@ from utils.util import load_model_from_config def load_speech_dvae(): return load_model_from_config("../experiments/train_diffusion_vocoder_22k_level.yml", - "dvae").cuda() + "dvae").cpu() def wav_to_mel(wav, mel_norms_file='../experiments/clips_mel_norms.pth'):