(
self,
architectures: list[str],
convert_type: ConvertType,
)
| 538 | return "embed" |
| 539 | |
| 540 | def _get_supported_pooling_tasks( |
| 541 | self, |
| 542 | architectures: list[str], |
| 543 | convert_type: ConvertType, |
| 544 | ) -> list[_ResolvedTask]: |
| 545 | registry = self.registry |
| 546 | |
| 547 | supported_tasks = list[_ResolvedTask]() |
| 548 | if registry.is_pooling_model(architectures, self) or convert_type in _RUNNER_CONVERTS["pooling"]: |
| 549 | supported_tasks.append("encode") |
| 550 | |
| 551 | extra_task = self._get_default_pooling_task(architectures) if convert_type == "none" else convert_type |
| 552 | supported_tasks.append(extra_task) |
| 553 | |
| 554 | return supported_tasks |
| 555 | |
| 556 | def _get_supported_tasks( |
| 557 | self, |
no test coverage detected