"vscode:/vscode.git/clone" did not exist on "6d2f74efb34566edc11477a651749173cac4c25d"
Unverified Commit ceab70c8 authored by Song Zhixin's avatar Song Zhixin Committed by GitHub
Browse files

[Bugfix] fix qwen3-asr response error (#33644)


Signed-off-by: default avatarjesse <szxfml@gmail.com>
Signed-off-by: default avatarCyrus Leung <cyrus.tl.leung@gmail.com>
Co-authored-by: default avatarCyrus Leung <cyrus.tl.leung@gmail.com>
parent 52683ccb
...@@ -125,6 +125,13 @@ class Qwen3ASRProcessingInfo(BaseProcessingInfo): ...@@ -125,6 +125,13 @@ class Qwen3ASRProcessingInfo(BaseProcessingInfo):
def get_supported_mm_limits(self) -> Mapping[str, int | None]: def get_supported_mm_limits(self) -> Mapping[str, int | None]:
return {"audio": None} return {"audio": None}
def get_data_parser(self) -> MultiModalDataParser:
feature_extractor = self.get_feature_extractor()
return Qwen3ASRMultiModalDataParser(
target_sr=feature_extractor.sampling_rate,
expected_hidden_size=self._get_expected_hidden_size(),
)
class Qwen3ASRDummyInputsBuilder(BaseDummyInputsBuilder[Qwen3ASRProcessingInfo]): class Qwen3ASRDummyInputsBuilder(BaseDummyInputsBuilder[Qwen3ASRProcessingInfo]):
def get_dummy_text(self, mm_counts: Mapping[str, int]) -> str: def get_dummy_text(self, mm_counts: Mapping[str, int]) -> str:
...@@ -194,12 +201,6 @@ class Qwen3ASRMultiModalDataParser(MultiModalDataParser): ...@@ -194,12 +201,6 @@ class Qwen3ASRMultiModalDataParser(MultiModalDataParser):
class Qwen3ASRMultiModalProcessor( class Qwen3ASRMultiModalProcessor(
Qwen3OmniMoeThinkerMultiModalProcessor, Qwen3OmniMoeThinkerMultiModalProcessor,
): ):
def _get_data_parser(self) -> MultiModalDataParser:
feature_extractor = self.info.get_feature_extractor()
return Qwen3ASRMultiModalDataParser(
target_sr=feature_extractor.sampling_rate,
)
def _get_mm_fields_config( def _get_mm_fields_config(
self, self,
hf_inputs: BatchFeature, hf_inputs: BatchFeature,
......
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