mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-04-18 17:27:06 +08:00
15 lines
381 B
Python
15 lines
381 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from dataclasses import dataclass
|
|
|
|
|
|
@dataclass
|
|
class AdapterMapping:
|
|
# Per every token in input_ids:
|
|
index_mapping: tuple[int, ...]
|
|
# Per sampled token:
|
|
prompt_mapping: tuple[int, ...]
|
|
|
|
def __post_init__(self):
|
|
self.index_mapping = tuple(self.index_mapping)
|
|
self.prompt_mapping = tuple(self.prompt_mapping) |