(
self,
architectures: list[str],
runner_type: RunnerType,
convert_type: ConvertType,
)
| 554 | return supported_tasks |
| 555 | |
| 556 | def _get_supported_tasks( |
| 557 | self, |
| 558 | architectures: list[str], |
| 559 | runner_type: RunnerType, |
| 560 | convert_type: ConvertType, |
| 561 | ) -> list[_ResolvedTask]: |
| 562 | if runner_type == "generate": |
| 563 | return self._get_supported_generation_tasks(architectures, convert_type) |
| 564 | if runner_type == "pooling": |
| 565 | return self._get_supported_pooling_tasks(architectures, convert_type) |
| 566 | |
| 567 | assert_never(runner_type) |
| 568 | |
| 569 | def _init_pooler_config(self) -> Optional["PoolerConfig"]: |
| 570 | if self.runner_type == "pooling": |
no test coverage detected