| 1234567891011121314 |
- from dataclasses import dataclass
- from typing import Tuple
- @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)
|
