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

more sampling fixes

Browse Source
This commit is contained in:
James Betker 2022-10-10 20:11:28 -06:00
parent 5d172fbf7e
commit f31a333c4f
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

@ -191,7 +191,7 @@ class AudioDiffusionFid(evaluator.Evaluator):
gen_wav = self.local_modules['vocoder'].inference(gen_mel)
real_dec = self.local_modules['vocoder'].inference(univnet_mel)
return gen_wav.float(), real_dec, SAMPLE_RATE
return gen_wav.float(), real_dec, gen_mel, univnet_mel, SAMPLE_RATE
def perform_diffusion_ctc(self, audio, codes, text):
SAMPLE_RATE = 24000