preprocess.py 30.3 KB
Newer Older
1
2
# SPDX-License-Identifier: Apache-2.0

3
import asyncio
4
5
from collections.abc import Mapping
from typing import Optional, Union, cast
6
7
8
9
10
11

from typing_extensions import assert_never

from vllm.config import ModelConfig
from vllm.logger import init_logger
from vllm.lora.request import LoRARequest
12
from vllm.multimodal import MULTIMODAL_REGISTRY, MultiModalRegistry
13
14
from vllm.multimodal.inputs import (MultiModalDataDict, MultiModalEncDecInputs,
                                    MultiModalInputs)
15
16
17
from vllm.prompt_adapter.request import PromptAdapterRequest
from vllm.transformers_utils.tokenizer_group import BaseTokenizerGroup

18
19
from .data import (DecoderOnlyInputs, EncoderDecoderInputs, ProcessorInputs,
                   PromptType, SingletonInputs, SingletonPrompt, token_inputs)
20
21
22
23
24
25
26
27
28
29
30
from .parse import is_explicit_encoder_decoder_prompt, parse_singleton_prompt

logger = init_logger(__name__)


class InputPreprocessor:

    def __init__(
        self,
        model_config: ModelConfig,
        tokenizer: Optional[BaseTokenizerGroup],
31
        mm_registry: MultiModalRegistry = MULTIMODAL_REGISTRY,
32
33
34
35
36
    ) -> None:
        super().__init__()

        self.model_config = model_config
        self.tokenizer = tokenizer
37
        self.mm_registry = mm_registry
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

    def get_tokenizer_group(self) -> BaseTokenizerGroup:
        if self.tokenizer is None:
            raise ValueError("You cannot pass text prompts when "
                             "`skip_tokenizer_init` is True")

        return self.tokenizer

    def get_bos_token_id(self,
                         lora_request: Optional[LoRARequest] = None
                         ) -> Optional[int]:
        if self.tokenizer is None:
            logger.warning("Using None for BOS token id because tokenizer "
                           "is not initialized")
            return None

        return self.tokenizer.get_lora_tokenizer(lora_request).bos_token_id

    def get_eos_token_id(self,
                         lora_request: Optional[LoRARequest] = None
                         ) -> Optional[int]:
        if self.tokenizer is None:
            logger.warning("Using None for EOS token id because tokenizer "
                           "is not initialized")
            return None

        return self.tokenizer.get_lora_tokenizer(lora_request).eos_token_id

    def get_decoder_start_token_id(self) -> Optional[int]:
        '''
        Obtain the decoder start token id employed by an encoder/decoder
        model. Returns None for non-encoder/decoder models or if the
        model config is unavailable.
        '''

73
        if not self.model_config.is_encoder_decoder:
74
75
76
            logger.warning_once(
                "Using None for decoder start token id because "
                "this is not an encoder/decoder model.")
77
78
79
            return None

        if (self.model_config is None or self.model_config.hf_config is None):
80
81
82
            logger.warning_once(
                "Using None for decoder start token id because "
                "model config is not available.")
83
84
85
86
87
            return None

        dec_start_token_id = getattr(self.model_config.hf_config,
                                     'decoder_start_token_id', None)
        if dec_start_token_id is None:
88
89
90
91
            logger.warning_once(
                "Falling back on <BOS> for decoder start token "
                "id because decoder start token id is not "
                "available.")
92
93
94
95
            dec_start_token_id = self.get_bos_token_id()

        return dec_start_token_id

96
    def _get_default_enc_dec_decoder_prompt(self) -> list[int]:
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
        '''
        Specifically for encoder/decoder models:
        generate a default decoder prompt for when
        the user specifies only the encoder prompt.

        Encoder/decoder models utilize the decoder
        prompt in different ways; as new models are
        added, it is intended that this function
        will be extended to produce differing
        default decoder prompts, depending on the
        model variety.

        Absent a special case, the default behavior
        of this method is to mirror the behavior of
        the HuggingFace (HF) GenerationMixin for a None
        decoder prompt, which is to employ a logit processor
        setting to force the first decoded token to be <BOS>.
        Here, this behavior is approximated by having the
        "default" decoder prompt be <BOS>.

        However, it is possible that in the future
118
        other models may have different or more
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
        complex logic for the default decoder prompt.
        This motivates having a special helper method
        for default decoder prompts.

        Returns:

        * prompt_token_ids
        '''

        bos_token_id = self.get_bos_token_id()
        assert bos_token_id is not None
        return [bos_token_id]

    def _prepare_decoder_input_ids_for_generation(
        self,
134
135
        decoder_input_ids: Optional[list[int]],
    ) -> list[int]:
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
        """
        Prepares `decoder_input_ids` for generation with encoder-decoder models.

        Based on

        https://github.com/huggingface/transformers/blob/
        4037a2b5b1278736e566aec12e169100275545ea/
        src/transformers/generation/utils.py

        specifically GenerationMixin._prepare_decoder_input_ids_for_generation()

        Arguments:

        * decoder_input_ids: input token ids to preprocess

        Returns:

        * Processed token list
        """

        decoder_start_token_id = self.get_decoder_start_token_id()
        assert decoder_start_token_id is not None

        if decoder_input_ids is None:
            # no decoder prompt input ->
            # use decoder_start_token_id as decoder_input_ids
            decoder_input_ids = self._get_default_enc_dec_decoder_prompt()

164
165
        if (len(decoder_input_ids) == 0
                or decoder_input_ids[0] != decoder_start_token_id):
166
167
168
169
170
171
            decoder_input_ids = [decoder_start_token_id] + decoder_input_ids

        return decoder_input_ids

    def _apply_prompt_adapter(
        self,
172
        prompt_token_ids: list[int],
173
        prompt_adapter_request: Optional[PromptAdapterRequest],
174
    ) -> list[int]:
175
176
177
178
179
180
181
182
183
184
185
186
        if prompt_adapter_request:
            prompt_token_ids = (
                [0] * prompt_adapter_request.prompt_adapter_num_virtual_tokens
                + prompt_token_ids)

        return prompt_token_ids

    def _tokenize_prompt(
        self,
        prompt: str,
        request_id: str,
        lora_request: Optional[LoRARequest],
187
    ) -> list[int]:
188
189
190
191
192
        """
        Apply the model's tokenizer to a text prompt, returning the
        corresponding token IDs.
        """
        tokenizer = self.get_tokenizer_group()
193
194
195
196
197
198
        add_special_tokens = None
        if self.model_config.hf_config.model_type == "whisper":
            # For Whisper, special tokens should be provided by the user based
            # on the task and language of their request. Also needed to avoid
            # appending an EOS token to the prompt which disrupts generation.
            add_special_tokens = False
199
200
201
202
203
204

        if (self.model_config.encoder_config is not None
                and self.model_config.encoder_config.get(
                    "do_lower_case", False)):
            prompt = prompt.lower()

205
206
        return tokenizer.encode(request_id=request_id,
                                prompt=prompt,
207
208
                                lora_request=lora_request,
                                add_special_tokens=add_special_tokens)
209
210
211
212
213
214

    async def _tokenize_prompt_async(
        self,
        prompt: str,
        request_id: str,
        lora_request: Optional[LoRARequest],
215
    ) -> list[int]:
216
217
        """Async version of :meth:`_tokenize_prompt`."""
        tokenizer = self.get_tokenizer_group()
218
219
220
221
222
223
224
225
226
227
228
        add_special_tokens = None
        if self.model_config.hf_config.model_type == "whisper":
            # For Whisper, special tokens should be provided by the user based
            # on the task and language of their request. Also needed to avoid
            # appending an EOS token to the prompt which disrupts generation.
            add_special_tokens = False
        return await tokenizer.encode_async(
            request_id=request_id,
            prompt=prompt,
            lora_request=lora_request,
            add_special_tokens=add_special_tokens)
229

230
231
232
233
234
235
236
237
238
239
    def _can_process_multimodal(self) -> bool:
        model_config = self.model_config

        if not model_config.is_multimodal_model:
            raise ValueError("Your model does not support multi-modal inputs")

        # Interim measure so we can handle models that have yet to be
        # updated to use the new multi-modal processor
        can_process_multimodal = self.mm_registry.has_processor(model_config)
        if not can_process_multimodal:
240
241
242
243
244
245
246
247
248
            from vllm.model_executor.models.registry import _VLLM_MODELS
            if not any(arch in _VLLM_MODELS
                       for arch in model_config.architectures):
                logger.warning_once(
                    "Your model uses the legacy input pipeline, which will be "
                    "removed in an upcoming release. "
                    "Please upgrade to the new multi-modal processing pipeline "
                    "(https://docs.vllm.ai/en/latest/design/mm_processing.html)"
                )
249
250
251
252
253

        return can_process_multimodal

    def _process_multimodal(
        self,
254
        prompt: Union[str, list[int]],
255
256
257
        mm_data: MultiModalDataDict,
        mm_processor_kwargs: Optional[Mapping[str, object]],
        lora_request: Optional[LoRARequest],
258
        return_mm_hashes: bool = False,
259
    ) -> MultiModalInputs:
260
261
262
263
        """
        Apply the model's multi-modal processor to a multi-modal prompt,
        returning the corresponding token IDs and metadata.
        """
264
265
266
267
268
269
270
271
        # At the moment on model (PrithviGeoSpatialMAE) requires to be
        # initialized without a tokenizer while using also multi-modal
        # input.
        if not self.tokenizer:
            tokenizer = None
        else:
            tokenizer_group = self.get_tokenizer_group()
            tokenizer = tokenizer_group.get_lora_tokenizer(lora_request)
272
273
274
275
276
277
278

        mm_processor = self.mm_registry.create_processor(
            self.model_config, tokenizer)

        if mm_processor_kwargs is None:
            mm_processor_kwargs = {}

279
280
        return mm_processor.apply(prompt, mm_data, mm_processor_kwargs,
                                  return_mm_hashes)
281
282
283

    async def _process_multimodal_async(
        self,
284
        prompt: Union[str, list[int]],
285
286
287
        mm_data: MultiModalDataDict,
        mm_processor_kwargs: Optional[Mapping[str, object]],
        lora_request: Optional[LoRARequest],
288
        return_mm_hashes: bool = False,
289
    ) -> MultiModalInputs:
290
        """Async version of :meth:`_process_multimodal`."""
291
292
293
294
295
296
297
298
299
        # At the moment on model (PrithviGeoSpatialMAE) requires to be
        # initialized without a tokenizer while using also multi-modal
        # input.
        if not self.tokenizer:
            tokenizer = None
        else:
            tokenizer_group = self.get_tokenizer_group()
            tokenizer = await tokenizer_group.get_lora_tokenizer_async(
                lora_request)
300
301
302
303
304
305

        mm_processor = self.mm_registry.create_processor(
            self.model_config, tokenizer)
        if mm_processor_kwargs is None:
            mm_processor_kwargs = {}

306
307
        return mm_processor.apply(prompt, mm_data, mm_processor_kwargs,
                                  return_mm_hashes)
308

309
    def _prompt_to_llm_inputs(
310
        self,
311
        prompt: SingletonPrompt,
312
313
        request_id: str,
        lora_request: Optional[LoRARequest] = None,
314
        return_mm_hashes: bool = False,
315
    ) -> SingletonInputs:
316
317
        """
        Extract the singleton inputs from a prompt.
318
319
320
321

        Arguments:

        * request_id
322
        * prompt: single encoder or decoder input prompt
323
        * lora_request: this is only valid for decoder prompts
324
        * return_mm_hashes: whether to return multimodal hashes
325
326
327

        Returns:

328
329
        * :class:`SingletonInputs` instance
        """
330
        parsed = parse_singleton_prompt(prompt)
331
332

        if parsed["type"] == "str":
333
            prompt_text = parsed["content"]
334
            prompt_token_ids = self._tokenize_prompt(
335
                prompt_text,
336
337
338
                request_id=request_id,
                lora_request=lora_request,
            )
339
340
341
342
343
344
345
346
347
348

            return token_inputs(
                prompt=prompt_text,
                prompt_token_ids=prompt_token_ids,
            )

        if parsed["type"] == "tokens":
            tokens_content = parsed["content"]

            prompt_token_ids = tokens_content["prompt_token_ids"]
349
            token_type_ids = tokens_content.get("token_type_ids")
350
351
352
            multi_modal_data = tokens_content.get("multi_modal_data")
            mm_processor_kwargs = tokens_content.get("mm_processor_kwargs")

353
354
355
356
357
358
            if multi_modal_data is not None and self._can_process_multimodal():
                return self._process_multimodal(
                    prompt_token_ids,
                    multi_modal_data,
                    mm_processor_kwargs,
                    lora_request=lora_request,
359
                    return_mm_hashes=return_mm_hashes,
360
361
                )

362
363
            return token_inputs(
                prompt_token_ids=prompt_token_ids,
364
                token_type_ids=token_type_ids,
365
366
367
368
369
370
371
372
                multi_modal_data=multi_modal_data,
                mm_processor_kwargs=mm_processor_kwargs,
            )

        if parsed["type"] == "text":
            text_content = parsed["content"]

            prompt_text = text_content["prompt"]
373
374
375
376
377
378
379
380
381
            multi_modal_data = text_content.get("multi_modal_data")
            mm_processor_kwargs = text_content.get("mm_processor_kwargs")

            if multi_modal_data is not None and self._can_process_multimodal():
                return self._process_multimodal(
                    prompt_text,
                    multi_modal_data,
                    mm_processor_kwargs,
                    lora_request=lora_request,
382
                    return_mm_hashes=return_mm_hashes,
383
384
                )

385
            prompt_token_ids = self._tokenize_prompt(
386
                prompt_text,
387
388
389
                request_id=request_id,
                lora_request=lora_request,
            )
390
391
392
393
394
395
396

            return token_inputs(
                prompt=prompt_text,
                prompt_token_ids=prompt_token_ids,
                multi_modal_data=multi_modal_data,
                mm_processor_kwargs=mm_processor_kwargs,
            )
397

398
        assert_never(parsed)
399

400
    async def _prompt_to_llm_inputs_async(
401
        self,
402
        prompt: SingletonPrompt,
403
404
        request_id: str,
        lora_request: Optional[LoRARequest] = None,
405
        return_mm_hashes: bool = False,
406
    ) -> SingletonInputs:
407
        """Async version of :meth:`_extract_prompt_components`."""
408
        parsed = parse_singleton_prompt(prompt)
409
410

        if parsed["type"] == "str":
411
            prompt_text = parsed["content"]
412
            prompt_token_ids = await self._tokenize_prompt_async(
413
                prompt_text,
414
415
416
                request_id=request_id,
                lora_request=lora_request,
            )
417
418
419
420
421
422
423
424
425
426
427
428
429

            return token_inputs(
                prompt=prompt_text,
                prompt_token_ids=prompt_token_ids,
            )

        if parsed["type"] == "tokens":
            tokens_content = parsed["content"]

            prompt_token_ids = tokens_content["prompt_token_ids"]
            multi_modal_data = tokens_content.get("multi_modal_data")
            mm_processor_kwargs = tokens_content.get("mm_processor_kwargs")

430
431
432
433
434
435
            if multi_modal_data is not None and self._can_process_multimodal():
                return await self._process_multimodal_async(
                    prompt_token_ids,
                    multi_modal_data,
                    mm_processor_kwargs,
                    lora_request=lora_request,
436
                    return_mm_hashes=return_mm_hashes,
437
438
                )

439
440
441
442
443
444
445
446
447
448
            return token_inputs(
                prompt_token_ids=prompt_token_ids,
                multi_modal_data=multi_modal_data,
                mm_processor_kwargs=mm_processor_kwargs,
            )

        if parsed["type"] == "text":
            text_content = parsed["content"]

            prompt_text = text_content["prompt"]
449
450
451
452
453
454
455
456
457
            multi_modal_data = text_content.get("multi_modal_data")
            mm_processor_kwargs = text_content.get("mm_processor_kwargs")

            if multi_modal_data is not None and self._can_process_multimodal():
                return await self._process_multimodal_async(
                    prompt_text,
                    multi_modal_data,
                    mm_processor_kwargs,
                    lora_request=lora_request,
458
                    return_mm_hashes=return_mm_hashes,
459
460
                )

461
            prompt_token_ids = await self._tokenize_prompt_async(
462
                prompt_text,
463
464
465
                request_id=request_id,
                lora_request=lora_request,
            )
466
467
468
469
470
471
472

            return token_inputs(
                prompt=prompt_text,
                prompt_token_ids=prompt_token_ids,
                multi_modal_data=multi_modal_data,
                mm_processor_kwargs=mm_processor_kwargs,
            )
473

474
        assert_never(parsed)
475
476
477

    def _build_enc_dec_llm_inputs(
        self,
478
479
        encoder_inputs: SingletonInputs,
        decoder_inputs: Optional[SingletonInputs],
480
    ) -> EncoderDecoderInputs:
481
482
        if (encoder_inputs["type"] == "token"
                or encoder_inputs["type"] == "multimodal"):
483
484
            pass
        else:
485
            assert_never(encoder_inputs)  # type: ignore[arg-type]
486
487

        if decoder_inputs is None:
488
489
490
491
492
493
494
495
496
            if self.model_config.hf_config.model_type == "whisper":
                # For Whisper models, the text prompt should go to the decoder.
                # If no explicit encoder/decoder inputs, then copy the prompt
                # from the encoder to the decoder. The encoder tokens are later
                # overridden by the audio features.
                dec_token_ids = encoder_inputs["prompt_token_ids"].copy()
            else:
                dec_token_ids = self._prepare_decoder_input_ids_for_generation(
                    None)
497
            decoder_inputs = token_inputs(dec_token_ids)
498
499
        elif (decoder_inputs["type"] == "token"
              or decoder_inputs["type"] == "multimodal"):
500
501
502
503
504
505
506
507
            dec_token_ids = self._prepare_decoder_input_ids_for_generation(
                decoder_inputs["prompt_token_ids"])
            decoder_inputs["prompt_token_ids"] = dec_token_ids

            if "multi_modal_data" in decoder_inputs:
                raise ValueError("Multi-modal decoder inputs of encoder-"
                                 "decoder models are not supported yet")
        else:
508
            assert_never(encoder_inputs)  # type: ignore[arg-type]
509

510
        return EncoderDecoderInputs(
511
512
            encoder=encoder_inputs,
            decoder=decoder_inputs,
513
514
        )

515
516
517
518
    def _separate_enc_dec_inputs_from_mm_processor_outputs(
        self,
        inputs: SingletonInputs,
        decoder_inputs_to_override: Optional[SingletonInputs] = None,
519
    ) -> tuple[SingletonInputs, SingletonInputs]:
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
        """
        For encoder/decoder models only:
        Separate Encoder/Decoder inputs from a MultiModalEncDecInputs
        """
        encoder_inputs: SingletonInputs
        decoder_inputs: SingletonInputs
        if inputs["type"] == "multimodal":
            # Multimodal data inputs
            assert ("encoder_prompt" in inputs
                    and "encoder_prompt_token_ids" in inputs)
            inputs = cast(MultiModalEncDecInputs, inputs)
            encoder_inputs = token_inputs(
                prompt=inputs["encoder_prompt"],
                prompt_token_ids=inputs["encoder_prompt_token_ids"],
            )
            if decoder_inputs_to_override is not None:
                decoder_inputs = MultiModalInputs(
                    type="multimodal",
                    prompt=decoder_inputs_to_override.get("prompt", ""),
                    prompt_token_ids=decoder_inputs_to_override[
                        "prompt_token_ids"],
                    mm_kwargs=inputs["mm_kwargs"],
                    mm_placeholders=inputs["mm_placeholders"],
                )
            else:
                decoder_inputs = MultiModalInputs(
                    type="multimodal",
                    prompt=inputs["prompt"],
                    prompt_token_ids=inputs["prompt_token_ids"],
                    mm_kwargs=inputs["mm_kwargs"],
                    mm_placeholders=inputs["mm_placeholders"],
                )
        elif inputs["type"] == "token":
            # Text-only inputs
            encoder_inputs = token_inputs(prompt="", prompt_token_ids=[])
            decoder_inputs = decoder_inputs_to_override or inputs
        else:
            assert_never(inputs)  # type: ignore[arg-type]
        return encoder_inputs, decoder_inputs

560
561
    def _process_encoder_decoder_prompt(
        self,
562
        prompt: PromptType,
563
        request_id: str,
564
    ) -> EncoderDecoderInputs:
565
        """
566
        For encoder/decoder models only:
567
        Process an input prompt into an :class:`EncoderDecoderInputs` instance.
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585

        There are two types of input prompts:
        singleton prompts which carry only the
        encoder prompt, and explicit encoder/decoder
        prompts which carry both the encoder and the
        decoder prompts as member variables.

        This function handles the following scenarios:
        * Singleton encoder prompt: extract encoder prompt
          token ids & infer default decoder prompt token ids
        * Explicit encoder/decoder prompt: extract encoder
          and decoder prompt token ids

        Note that for Explicit encoder/decoder prompts,
        each sub-prompt (encoder or decoder prompt) can
        have any possible singleton type; thus this
        method relies on helper functions to obtain
        token ids for the sub-prompts.
586

587
588
        Arguments:

589
        * prompt: an input prompt
590
591
592
593
        * request_id

        Returns:

594
        * :class:`EncoderDecoderInputs` instance
595
        """
596
597
        encoder_inputs: SingletonInputs
        decoder_inputs: Optional[SingletonInputs]
598

599
        if is_explicit_encoder_decoder_prompt(prompt):
600
            encoder_inputs = self._prompt_to_llm_inputs(
601
                prompt["encoder_prompt"],
602
603
                request_id=request_id,
            )
604
            if (decoder_input := prompt["decoder_prompt"]) is None:
605
                decoder_inputs = None
606
            else:
607
                decoder_inputs = self._prompt_to_llm_inputs(
608
609
610
                    decoder_input,
                    request_id=request_id,
                )
611
612
613
614
615
616
617
            # For multimodal model, override decoder prompt from processor
            # with explicit decoder prompt.
            if self.model_config.is_multimodal_model and (
                    self._can_process_multimodal()):
                encoder_inputs, decoder_inputs = (
                    self._separate_enc_dec_inputs_from_mm_processor_outputs(
                        encoder_inputs, decoder_inputs))
618
        else:
619
            inputs = self._prompt_to_llm_inputs(
620
                prompt,
621
622
                request_id=request_id,
            )
623
624
625
626
627
628
629
630
            if self.model_config.is_multimodal_model and (
                    self._can_process_multimodal()):
                # Encoder-Decoder Multimodal model
                encoder_inputs, decoder_inputs = (
                    self._separate_enc_dec_inputs_from_mm_processor_outputs(
                        inputs))
            else:
                encoder_inputs = inputs
631

632
                decoder_inputs = None
633
634

        return self._build_enc_dec_llm_inputs(encoder_inputs, decoder_inputs)
635
636
637

    async def _process_encoder_decoder_prompt_async(
        self,
638
        prompt: PromptType,
639
        request_id: str,
640
    ) -> EncoderDecoderInputs:
641
        """Async version of :meth:`_process_encoder_decoder_prompt`."""
642
643
        encoder_inputs: SingletonInputs
        decoder_inputs: Optional[SingletonInputs]
644

645
        if is_explicit_encoder_decoder_prompt(prompt):
646
            encoder_task = self._prompt_to_llm_inputs_async(
647
                prompt["encoder_prompt"],
648
649
650
                request_id=request_id,
            )

651
            if (decoder_input := prompt["decoder_prompt"]) is None:
652
653
                encoder_inputs = await encoder_task
                decoder_inputs = None
654
            else:
655
                decoder_task = self._prompt_to_llm_inputs_async(
656
657
658
659
                    decoder_input,
                    request_id=request_id,
                )

660
                encoder_inputs, decoder_inputs = await asyncio.gather(
661
                    encoder_task, decoder_task)
662
663
664
665
666
667
668
669

            # For multimodal model, override decoder prompt from processor
            # with explicit decoder prompt.
            if self.model_config.is_multimodal_model and (
                    self._can_process_multimodal()):
                encoder_inputs, decoder_inputs = (
                    self._separate_enc_dec_inputs_from_mm_processor_outputs(
                        encoder_inputs, decoder_inputs))
670
        else:
671
            inputs = await self._prompt_to_llm_inputs_async(
672
                prompt,
673
674
                request_id=request_id,
            )
675
676
677
678
679
680
681
682
            if self.model_config.is_multimodal_model and (
                    self._can_process_multimodal()):
                # Encoder-Decoder Multimodal model
                encoder_inputs, decoder_inputs = (
                    self._separate_enc_dec_inputs_from_mm_processor_outputs(
                        inputs))
            else:
                encoder_inputs = inputs
683

684
                decoder_inputs = None
685
686

        return self._build_enc_dec_llm_inputs(encoder_inputs, decoder_inputs)
687
688
689

    def _build_decoder_only_llm_inputs(
        self,
690
        prompt_inputs: DecoderOnlyInputs,
691
        prompt_adapter_request: Optional[PromptAdapterRequest],
692
    ) -> DecoderOnlyInputs:
693
694
        if (prompt_inputs["type"] == "token"
                or prompt_inputs["type"] == "multimodal"):
695
696
697
698
699
            prompt_inputs["prompt_token_ids"] = self._apply_prompt_adapter(
                prompt_inputs["prompt_token_ids"],
                prompt_adapter_request=prompt_adapter_request,
            )
        else:
700
            assert_never(prompt_inputs)  # type: ignore[arg-type]
701

702
        return prompt_inputs
703
704
705

    def _process_decoder_only_prompt(
        self,
706
        prompt: SingletonPrompt,
707
708
709
        request_id: str,
        lora_request: Optional[LoRARequest] = None,
        prompt_adapter_request: Optional[PromptAdapterRequest] = None,
710
        return_mm_hashes: bool = False,
711
    ) -> DecoderOnlyInputs:
712
        """
713
        For decoder-only models:
714
        Process an input prompt into an :class:`DecoderOnlyInputs` instance.
715
716
717

        Arguments:

718
        * prompt: input prompt
719
720
721
        * request_id
        * lora_request
        * prompt_adapter_request
722
        * return_mm_hashes
723
724
725

        Returns:

726
        * :class:`DecoderOnlyInputs` instance
727
        """
728

729
        prompt_comps = self._prompt_to_llm_inputs(
730
            prompt,
731
732
            request_id=request_id,
            lora_request=lora_request,
733
            return_mm_hashes=return_mm_hashes,
734
735
736
737
738
739
740
741
742
        )

        return self._build_decoder_only_llm_inputs(
            prompt_comps,
            prompt_adapter_request=prompt_adapter_request,
        )

    async def _process_decoder_only_prompt_async(
        self,
743
        prompt: SingletonPrompt,
744
745
746
        request_id: str,
        lora_request: Optional[LoRARequest] = None,
        prompt_adapter_request: Optional[PromptAdapterRequest] = None,
747
        return_mm_hashes: bool = False,
748
    ) -> DecoderOnlyInputs:
749
        """Async version of :meth:`_process_decoder_only_prompt`."""
750
        prompt_comps = await self._prompt_to_llm_inputs_async(
751
            prompt,
752
753
            request_id=request_id,
            lora_request=lora_request,
754
            return_mm_hashes=return_mm_hashes,
755
756
757
758
759
760
761
762
763
        )

        return self._build_decoder_only_llm_inputs(
            prompt_comps,
            prompt_adapter_request=prompt_adapter_request,
        )

    def preprocess(
        self,
764
        prompt: PromptType,
765
766
767
        request_id: str,
        lora_request: Optional[LoRARequest] = None,
        prompt_adapter_request: Optional[PromptAdapterRequest] = None,
768
        return_mm_hashes: bool = False,
769
    ) -> ProcessorInputs:
770
        """Preprocess the input prompt."""
771
        if self.model_config.is_encoder_decoder:
772
773
774
            assert not return_mm_hashes, (
                "Multimodal hashes for encoder-decoder models should not be ",
                "returned until they are supported on vLLM V1.")
775
776
777
            # Encoder-decoder model requires special mapping of
            # input prompts to encoder & decoder
            return self._process_encoder_decoder_prompt(
778
                prompt,
779
780
781
                request_id=request_id,
            )

782
        if is_explicit_encoder_decoder_prompt(prompt):
783
784
785
786
787
            raise ValueError("Cannot pass encoder-decoder prompt "
                             "to decoder-only models")

        # Decoder-only operation
        return self._process_decoder_only_prompt(
788
            prompt,
789
790
791
            request_id=request_id,
            lora_request=lora_request,
            prompt_adapter_request=prompt_adapter_request,
792
            return_mm_hashes=return_mm_hashes,
793
794
795
796
        )

    async def preprocess_async(
        self,
797
        prompt: PromptType,
798
799
800
        request_id: str,
        lora_request: Optional[LoRARequest] = None,
        prompt_adapter_request: Optional[PromptAdapterRequest] = None,
801
        return_mm_hashes: bool = False,
802
    ) -> ProcessorInputs:
803
        """Async version of :meth:`preprocess`."""
804
        if self.model_config.is_encoder_decoder:
805
806
807
            assert not return_mm_hashes, (
                "Multimodal hashes for encoder-decoder models should not be ",
                "returned until they are supported on vLLM V1.")
808
809
810
            # Encoder-decoder model requires special mapping of
            # input prompts to encoder & decoder
            return await self._process_encoder_decoder_prompt_async(
811
                prompt,
812
813
814
                request_id=request_id,
            )

815
        if is_explicit_encoder_decoder_prompt(prompt):
816
817
818
819
820
            raise ValueError("Cannot pass encoder-decoder prompt "
                             "to decoder-only models")

        # Decoder-only operation
        return await self._process_decoder_only_prompt_async(
821
            prompt,
822
823
824
            request_id=request_id,
            lora_request=lora_request,
            prompt_adapter_request=prompt_adapter_request,
825
            return_mm_hashes=return_mm_hashes,
826
        )