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

Fix eval

Browse Source
This commit is contained in:
James Betker 2022-10-10 14:22:36 -06:00
parent 9502e0755e
commit 5d172fbf7e
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/trainer/eval/audio_diffusion_fid.py
View File

@ -170,8 +170,7 @@ class AudioDiffusionFid(evaluator.Evaluator):
torch.tensor([text_codes.shape[-1]], device=mel.device), torch.tensor([text_codes.shape[-1]], device=mel.device),
mel_codes, mel_codes,
torch.tensor([mel_codes.shape[-1]*mlc], device=mel.device), torch.tensor([mel_codes.shape[-1]*mlc], device=mel.device),
text_first=True, raw_mels=None, return_latent=True, text_first=True, raw_mels=None, return_latent=True)
clip_inputs=False)
return auto_latents return auto_latents
def tts9_get_dvae_codes(self, mel, text): def tts9_get_dvae_codes(self, mel, text):