This commit is contained in:
James Betker 2022-10-10 14:22:36 -06:00
parent 9502e0755e
commit 5d172fbf7e

View File

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