Skip to content

GitLab

Unverified Commit 64ec802e authored by Yusong Wu's avatar Yusong Wu Committed by GitHub
Browse files

fix bug of CLAP dataloader (#22674) 

fix bug of CLAP: https://github.com/LAION-AI/CLAP/issues/62
parent 3db2e404
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/models/clap/feature_extraction_clap.py
View file @ 64ec802e
......@@ -194,7 +194,7 @@ class ClapFeatureExtractor(SequenceFeatureExtractor):
mel, size=[chunk_frames, 64], mode="bilinear", align_corners=False, antialias=False
)
mel_shrink = mel_shrink[0][0].numpy()
mel_fusion = np.stack([mel_chunk_front, mel_chunk_middle, mel_chunk_back, mel_shrink], axis=0)
mel_fusion = np.stack([mel_shrink, mel_chunk_front, mel_chunk_middle, mel_chunk_back], axis=0)
return mel_fusion
def _get_input_mel(self, waveform: np.array, max_length, truncation, padding) -> np.array:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment