Are there any updates or solutions yet?
I am having a similar trouble loading a model.
Edit: I don’t think that this should be marked as solved, since the solution refers to a pytorch version imcompatibility. This sounds more like a workaround than a solution. To me it should be compatible with multiple versions of pytorch since not everybody can up- or downgrade their version.