riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

[MusicGen] Fix audio channel attribute (#27440)

Browse Source 
[MusicGen] Fix mono logit test
This commit is contained in:
Sanchit Gandhi 2023-12-01 17:10:03 +00:00 committed by GitHub
parent abd4cbd775
commit 2c658b5a42
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/musicgen/modeling_musicgen.py
View File

@ -1869,7 +1869,7 @@ class MusicgenForConditionalGeneration(PreTrainedModel):
"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)