fix asr ut (#501)

opea-project · Aug 16, 2024 · 9580298 · 9580298
1 parent fecf4ac
commit 9580298
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/comps/asr/whisper/whisper_model.py b/comps/asr/whisper/whisper_model.py
@@ -148,7 +148,7 @@ def audio2text(self, audio_path):
                 return_tensors="pt",
                 sampling_rate=16000,
             )
-        elif self.device == "hpu":
+        elif self.device == "hpu" and processed_inputs.input_features.shape[-1] > 3000:
             processed_inputs["input_features"] = torch.nn.functional.pad(
                 processed_inputs.input_features,
                 (0, self.hpu_max_len - processed_inputs.input_features.size(-1)),