Unable to export diar_msdd_telephonic (neural diarizer) model to onnx format

Describe the bug

I am trying to export the diarization MSDD telephonic model to onnx format but i get the following error:

File “/home/chidalgo/.local/lib/python3.10/site-packages/nemo/core/classes/exportable.py”, line 113, in export
out, descr, out_example = model._export(
File “/home/chidalgo/.local/lib/python3.10/site-packages/nemo/core/classes/exportable.py”, line 177, in _export
input_example = self.input_module.input_example()
File “/home/chidalgo/.local/lib/python3.10/site-packages/torch/nn/modules/module.py”, line 1695, in getattr
raise AttributeError(f"‘{type(self).name}’ object has no attribute ‘{name}’")
AttributeError: ‘MyExportableModel’ object has no attribute ‘input_module’. Did you mean: ‘output_module’?.

Steps/Code to reproduce bug

Here is the code i used.

import nemo
from nemo.collections.asr.models import EncDecDiarLabelModel
from nemo.core.classes import ModelPT, Exportable

class MyExportableModel(EncDecDiarLabelModel, ModelPT, Exportable):

sd_model = MyExportableModel.from_pretrained(model_name = “diar_msdd_telephonic”)


Expected behavior

To get a file named sd_model.onnx with the model in onnx format

Environment overview (please complete the following information)

  • Environment location: [Bare-metal]
  • Method of NeMo install: pip install nemo_toolkit[‘all’]

Environment details

If NVIDIA docker image is used you don’t need to specify these.
Otherwise, please provide:

  • Ubuntu 20.04
  • 2.1.2
  • Python 3.10.12

Additional context

i also tried with the export.py script but i get the following:

File “/home/chidalgo/.local/lib/python3.10/site-packages/nemo/core/classes/common.py”, line 507, in from_config_dict
raise e
File “/home/chidalgo/.local/lib/python3.10/site-packages/nemo/core/classes/common.py”, line 499, in from_config_dict
instance = cls(cfg=config, trainer=trainer)
TypeError: Can’t instantiate abstract class ModelPT with abstract methods list_available_models, setup_training_data, setup_validation_data

thank you, i have seen other issues with the same bug but it has not been solved