Initialization of register_buffer for loading model

[thomas]

Oh, ok
This is exactly what I needed !
Thank you

(Yes I did a typo in the minimal working example)