Skip to content

GitLab

Unverified Commit 00d3310d authored by Cyrus Leung's avatar Cyrus Leung Committed by GitHub
Browse files

[Bugfix] Update Ultravox compatibility (#29588) 


Signed-off-by: default avatarDarkLight1337 <tlleungac@connect.ust.hk>
parent da3222f3
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
vllm/model_executor/models/ultravox.py
View file @ 00d3310d
...@@ -116,7 +116,12 @@ class UltravoxProcessingInfo(BaseProcessingInfo): ...@@ -116,7 +116,12 @@ class UltravoxProcessingInfo(BaseProcessingInfo):
def get_feature_extractor(self, **kwargs: object) -> WhisperFeatureExtractor: def get_feature_extractor(self, **kwargs: object) -> WhisperFeatureExtractor:
hf_processor = self.get_hf_processor(**kwargs) hf_processor = self.get_hf_processor(**kwargs)
# Changed in https://huggingface.co/fixie-ai/ultravox-v0_5-llama-3_2-1b/commit/9a3c571b8fdaf1e66dd3ea61bbcb6db5c70a438e
audio_processor = hf_processor.audio_processor # type: ignore audio_processor = hf_processor.audio_processor # type: ignore
if isinstance(audio_processor, WhisperFeatureExtractor):
return audio_processor
feature_extractor = audio_processor.feature_extractor # type: ignore feature_extractor = audio_processor.feature_extractor # type: ignore
assert isinstance(feature_extractor, WhisperFeatureExtractor) assert isinstance(feature_extractor, WhisperFeatureExtractor)
return feature_extractor return feature_extractor
......
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