mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Load dvae to cpu

Browse Source
This commit is contained in:
James Betker 2022-02-23 21:21:45 -07:00
parent 81017d9696
commit e6824e398f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/scripts/audio/gen/speech_synthesis_utils.py
View File

@ -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'):