add remapping to safetensors.torch.load_model

[f-fuchs]

Feature request

Add a way to remap model weights.

Motivation

At the moment it is not possible to load the weights of a compiled model into the uncompiled model, even though the only difference is that for the compiled model the weights are prefixed with "_orig_mod". Maybe this can be remapped internally.

Your contribution

not sure.