(self)
| 567 | assert_never(runner_type) |
| 568 | |
| 569 | def _init_pooler_config(self) -> Optional["PoolerConfig"]: |
| 570 | if self.runner_type == "pooling": |
| 571 | if isinstance(self.override_pooler_config, dict): |
| 572 | self.override_pooler_config = PoolerConfig(**self.override_pooler_config) |
| 573 | |
| 574 | pooler_config = self.override_pooler_config or PoolerConfig() |
| 575 | |
| 576 | base_config = get_pooling_config(self.model, self.revision) |
| 577 | if base_config is not None: |
| 578 | for k, v in base_config.items(): |
| 579 | if getattr(pooler_config, k) is None: |
| 580 | setattr(pooler_config, k, v) |
| 581 | |
| 582 | default_pooling_type = self._model_info.default_pooling_type |
| 583 | if pooler_config.pooling_type is None: |
| 584 | pooler_config.pooling_type = default_pooling_type |
| 585 | |
| 586 | return pooler_config |
| 587 | |
| 588 | return None |
| 589 | |
| 590 | def _get_download_model(self, model_name, model_type="default"): |
| 591 | # TODO: Provide dynamic graph for self-downloading and save to the specified download directory. |
no test coverage detected