(model_cfg: ConfigDict)
| 13 | |
| 14 | |
| 15 | def build_model_from_cfg(model_cfg: ConfigDict): |
| 16 | model_cfg = copy.deepcopy(model_cfg) |
| 17 | model_cfg.pop('run_cfg', None) |
| 18 | model_cfg.pop('max_out_len', None) |
| 19 | model_cfg.pop('batch_size', None) |
| 20 | model_cfg.pop('abbr', None) |
| 21 | model_cfg.pop('summarizer_abbr', None) |
| 22 | model_cfg.pop('pred_postprocessor', None) |
| 23 | model_cfg.pop('min_out_len', None) |
| 24 | return MODELS.build(model_cfg) |