From 951f3718cbba63f330a8a4110f9a130c73405297 Mon Sep 17 00:00:00 2001 From: wangzeyuan Date: Wed, 23 Jul 2025 16:49:03 +0800 Subject: [PATCH] fix whisper feature bug --- finetune_codes/modeling_kimia.py | 1 - 1 file changed, 1 deletion(-) diff --git a/finetune_codes/modeling_kimia.py b/finetune_codes/modeling_kimia.py index 00ff3c4..ce22b04 100644 --- a/finetune_codes/modeling_kimia.py +++ b/finetune_codes/modeling_kimia.py @@ -672,7 +672,6 @@ def forward( audio_emb = self.embed_tokens(input_ids) if self.use_whisper_feature and whisper_input_feature is not None: if not isinstance(whisper_input_feature, list): - whisper_input_feature = whisper_input_feature.squeeze(0) whisper_input_feature = [whisper_input_feature] media_start_idx = (input_ids == self.kimia_media_begin).nonzero()