TypeError: RNNTDecoderJoint.forward() missing 3 required positional arguments: 'target_length', 'input_states_1', and 'input_states_2'

[mehadi92]

When I'm performing nemo2riva conversion I'm getting this error. I'm using
nemo=1.23.0 (my .nemo model was trained with this version)
nemo2riva=2.16.0 with --onnx-opset=17

[NeMo E 2024-10-28 06:40:46 cookbook:132] ERROR: Export failed. Please make sure your NeMo model class (<class 'nemo.collections.asr.models.hybrid_rnnt_ctc_bpe_models.EncDecHybridRNNTCTCBPEModel'>) has working export() and that you have the latest NeMo package installed with [all] dependencies.
Traceback (most recent call last):
  File "/tmp/tmp.f3FLAUYEi5/miniconda/envs/nemo/bin/nemo2riva", line 8, in <module>
    sys.exit(nemo2riva())
  File "/tmp/tmp.f3FLAUYEi5/miniconda/envs/nemo/lib/python3.10/site-packages/nemo2riva/cli/nemo2riva.py", line 49, in nemo2riva
    Nemo2Riva(args)
  File "/tmp/tmp.f3FLAUYEi5/miniconda/envs/nemo/lib/python3.10/site-packages/nemo2riva/convert.py", line 87, in Nemo2Riva
    export_model(
  File "/tmp/tmp.f3FLAUYEi5/miniconda/envs/nemo/lib/python3.10/site-packages/nemo2riva/cookbook.py", line 133, in export_model
    raise e
  File "/tmp/tmp.f3FLAUYEi5/miniconda/envs/nemo/lib/python3.10/site-packages/nemo2riva/cookbook.py", line 88, in export_model
    _, descriptions = model.export(
  File "/tmp/tmp.f3FLAUYEi5/miniconda/envs/nemo/lib/python3.10/site-packages/nemo/core/classes/exportable.py", line 114, in export
    out, descr, out_example = model._export(
  File "/tmp/tmp.f3FLAUYEi5/miniconda/envs/nemo/lib/python3.10/site-packages/nemo/core/classes/exportable.py", line 192, in _export
    output_example = tuple(self.forward(*input_list, **input_dict))
TypeError: RNNTDecoderJoint.forward() missing 3 required positional arguments: 'target_length', 'input_states_1', and 'input_states_2'