File tree 1 file changed +8
-1
lines changed
vllm/model_executor/models
1 file changed +8
-1
lines changed Original file line number Diff line number Diff line change @@ -150,8 +150,15 @@ def _call_hf_processor(
150
150
mm_data : Mapping [str , object ],
151
151
mm_kwargs : Mapping [str , Any ],
152
152
) -> BatchFeature :
153
+ # NOTE - we rename audios -> audio in mm data because transformers has
154
+ # deprecated audios for the qwen2audio processor and will remove
155
+ # support for it in transformers 4.54.
156
+ audios = mm_data .pop ("audios" , [])
157
+ if audios :
158
+ mm_data ["audio" ] = audios
159
+
153
160
# Text-only input not supported in composite processor
154
- if not mm_data .get ("audios " , []):
161
+ if not mm_data .get ("audio " , []):
155
162
prompt_ids = self .info .get_tokenizer ().encode (prompt )
156
163
prompt_ids = self ._apply_hf_processor_tokens_only (prompt_ids )
157
164
return BatchFeature (dict (input_ids = [prompt_ids ]), tensor_type = "pt" )
You can’t perform that action at this time.
0 commit comments