Model Export Error in S3PRL

[Hitori-de]

I got an error trying to export my model into ONNX

Traceback (most recent call last):
  File "convertmodel.py", line 4, in <module>
    m.export_from_zip(
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/export_asr.py", line 197, in export_from_zip
    self.export(model, tag_name, quantize, optimize)
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/export_asr.py", line 78, in export
    enc_model = get_encoder(
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/models/__init__.py", line 60, in get_encoder
    return DefaultEncoder(_model, frontend, **export_config)
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/models/encoder_wrapper.py", line 18, in __init__
    self.get_frontend(kwargs)
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/models/encoder_wrapper.py", line 29, in get_frontend
    self.frontend_model = get_frontend_models(self.frontend, kwargs)
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/models/__init__.py", line 81, in get_frontend_models
    return S3PRLModel(model, **export_config)
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/models/frontends/s3prl.py", line 143, in __init__
    self.upstream = HubertModel(model.upstream, max_seq_len)
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/models/frontends/s3prl.py", line 77, in __init__
    raise RuntimeError(
RuntimeError: S3PRL version is not supported. Please install v0.3 or v0.4

I downgraded my s3prl version from v0.4.17 -> v0.4.0 due to unsupported version error then i got this error after that.

Traceback (most recent call last):
  File "convertmodel.py", line 4, in <module>
    m.export_from_zip(
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet_onnx/export/asr/export_asr.py", line 205, in export_from_zip
    model = Speech2Text(**model_config)
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/typeguard/__init__.py", line 1015, in wrapper
    retval = func(*args, **kwargs)
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet2/bin/asr_inference.py", line 147, in __init__
    asr_model, asr_train_args = task.build_model_from_file(
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/typeguard/__init__.py", line 1015, in wrapper
    retval = func(*args, **kwargs)
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet2/tasks/abs_task.py", line 2329, in build_model_from_file
    model = cls.build_model(args)
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/typeguard/__init__.py", line 1015, in wrapper
    retval = func(*args, **kwargs)
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet2/tasks/asr.py", line 538, in build_model
    frontend = frontend_class(**args.frontend_conf)
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/typeguard/__init__.py", line 1015, in wrapper
    retval = func(*args, **kwargs)
  File "/home/lwobole/Thesis/espnet/tools/miniconda/envs/espnet/lib/python3.8/site-packages/espnet2/asr/frontend/s3prl.py", line 47, in __init__
    upstream = S3PRLUpstream(
TypeError: __init__() got an unexpected keyword argument 'normalize'

I tried commenting out that line and the lines after that since it still showed an error. Is this due to a broken installation? or version mismatch among frameworks?
I have Espnet v202503 and S3PRL v0.4.17 installed
Please add support for latest versions of Espnet and S3PRL. Thank you!