Could you remove the direct loading of the obs_model and use the recommended workflow of creating the model instance and loading its state_dict afterwards?
If you get stuck, feel free to provide a minimal, executable code snippet to reproduce the issue.