From 2da8f8a666da049393fb23060abf8db9991a3467 Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 23 May 2022 07:06:25 -0600 Subject: [PATCH] fmp --- codes/scripts/audio/gen/use_mel2vec_codes.py | 1 + 1 file changed, 1 insertion(+) diff --git a/codes/scripts/audio/gen/use_mel2vec_codes.py b/codes/scripts/audio/gen/use_mel2vec_codes.py index 207ab1bc..74b8a783 100644 --- a/codes/scripts/audio/gen/use_mel2vec_codes.py +++ b/codes/scripts/audio/gen/use_mel2vec_codes.py @@ -32,6 +32,7 @@ if __name__ == '__main__': mel = TorchMelSpectrogramInjector({'n_mel_channels': 256, 'mel_fmax': 22000, 'normalize': True, 'in': 'in', 'out': 'out'}, {})({'in': wav.unsqueeze(0)})['out'] codes = model.get_codes(mel) + codes2 = model.get_codes(mel) collapsed = collapse_codegroups(codes) recovered = recover_codegroups(collapsed, 8)