File size: 683 Bytes
d91766b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | from diffulex.config import Config
from diffulex.engine.request import AutoReq
from diffulex.sampling_params import SamplingParams
from diffulex.strategy_template.multi_block.engine.request import MultiBlockReqTemplate
@AutoReq.register("d2f", is_default=True)
class D2fReq(MultiBlockReqTemplate):
"""Req for D2F strategy. Accepts config for AutoReq.create()."""
def __init__(
self,
token_ids: list[int],
sampling_params: SamplingParams = SamplingParams(),
config: Config | None = None,
):
super().__init__(token_ids, sampling_params)
# config is passed by AutoReq.create(); req attributes are set in init_multi_block()
|