missing 1 required positional argument: 'output_dim'

[bulatef]

I keep getting this error "TypeError: init() missing 1 required positional argument: 'output_dim'" when I load the model (DETIRE_model.h5) for testing.
I'm running the code on a conda environment with python=3.6.5 and keras=1.2.0 just like the requirements.
I believe the model was trained on a different version or something so is it possible to export your conda environment here so we can import it and use the same versions you used for training?