(self)
| 87 | raise TypeError("Wrong type for `config`, must be subclass of BaseConfig") |
| 88 | |
| 89 | def create_llm(self) -> Callable[[...], List[str]]: |
| 90 | if self.config.seed: |
| 91 | set_seed(self.config.seed) |
| 92 | engine, sampling_params = llm_engine(self.config) |
| 93 | self.engine = engine |
| 94 | self.generate_sampling_params = sampling_params |
| 95 | self.value_sampling_params = copy.deepcopy(sampling_params) |
| 96 | self.value_sampling_params.max_tokens = 1 |
| 97 | self.value_sampling_params.n = 1 |
| 98 | return partial( |
| 99 | local_generator, |
| 100 | engine=self.engine, |
| 101 | ) |
| 102 | |
| 103 | @staticmethod |
| 104 | def processor(solver: BaseTree, output: List[RequestOutput]) -> BaseTree: |
no test coverage detected