(
self,
architectures: list[str],
)
| 524 | return supported_tasks |
| 525 | |
| 526 | def _get_default_pooling_task( |
| 527 | self, |
| 528 | architectures: list[str], |
| 529 | ) -> Literal["embed"]: |
| 530 | # Temporarily does not support classification and reward. |
| 531 | for arch in architectures: |
| 532 | match = try_match_architecture_defaults(arch, runner_type="pooling") |
| 533 | if match: |
| 534 | _, (_, convert_type) = match |
| 535 | assert convert_type != "none" |
| 536 | return convert_type |
| 537 | |
| 538 | return "embed" |
| 539 | |
| 540 | def _get_supported_pooling_tasks( |
| 541 | self, |
no test coverage detected