(
self,
architectures: list[str],
convert_type: ConvertType,
)
| 510 | return convert_type |
| 511 | |
| 512 | def _get_supported_generation_tasks( |
| 513 | self, |
| 514 | architectures: list[str], |
| 515 | convert_type: ConvertType, |
| 516 | ) -> list[_ResolvedTask]: |
| 517 | registry = self.registry |
| 518 | |
| 519 | supported_tasks = list[_ResolvedTask]() |
| 520 | if registry.is_text_generation_model(architectures, self) or convert_type in _RUNNER_CONVERTS["generate"]: |
| 521 | supported_tasks.append("generate") |
| 522 | |
| 523 | # TODO:Temporarily does not support transcription. |
| 524 | return supported_tasks |
| 525 | |
| 526 | def _get_default_pooling_task( |
| 527 | self, |
no test coverage detected