kxngluca/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

Load diffusion_fid DVAE into the correct cuda device

Browse Source
This commit is contained in:
James Betker 2022-03-04 13:42:14 -07:00
parent e1052a5e32
commit 382681a35d
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/trainer/eval/audio_diffusion_fid.py
View File

@ -48,7 +48,7 @@ class AudioDiffusionFid(evaluator.Evaluator):
if mode == 'tts': if mode == 'tts':
self.diffusion_fn = self.perform_diffusion_tts self.diffusion_fn = self.perform_diffusion_tts
elif mode == 'vocoder': elif mode == 'vocoder':
self.dvae = load_speech_dvae() self.dvae = load_speech_dvae().to(self.env['device'])
self.dvae.eval() self.dvae.eval()
self.diffusion_fn = self.perform_diffusion_vocoder self.diffusion_fn = self.perform_diffusion_vocoder