From fed502536f1fb045272f7a5a256a4f585918b938 Mon Sep 17 00:00:00 2001 From: sanchit-gandhi Date: Fri, 10 Nov 2023 17:39:16 +0000 Subject: [PATCH] [MusicGen] Fix mono logit test --- src/transformers/models/musicgen/modeling_musicgen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/musicgen/modeling_musicgen.py b/src/transformers/models/musicgen/modeling_musicgen.py index 584b29e62313..8cca8108efd0 100644 --- a/src/transformers/models/musicgen/modeling_musicgen.py +++ b/src/transformers/models/musicgen/modeling_musicgen.py @@ -1869,7 +1869,7 @@ def forward( "disabled by setting `chunk_length=None` in the audio encoder." ) - if self.config.audio_channels == 2 and audio_codes.shape[2] == self.decoder.num_codebooks // 2: + if self.config.decoder.audio_channels == 2 and audio_codes.shape[2] == self.decoder.num_codebooks // 2: # mono input through encodec that we convert to stereo audio_codes = audio_codes.repeat_interleave(2, dim=2)