"vllm/vscode:/vscode.git/clone" did not exist on "b2789112294cb1f54c55c2011d2a5737c2b22cae"
parse.py 396 Bytes
Newer Older
1
# SPDX-License-Identifier: Apache-2.0
2
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3

4
from .data import ProcessorInputs, SingletonInputs
5
6


7
8
def split_enc_dec_inputs(
    inputs: ProcessorInputs,
9
) -> tuple[SingletonInputs | None, SingletonInputs]:
10
11
    if inputs["type"] == "enc_dec":
        return inputs["encoder_prompt"], inputs["decoder_prompt"]
12
13

    return None, inputs