Have a look at the other init implementations: code 3.6k.
It’s basically a vanilla Python function setting the values.
Just create a new function with your proprietary method and call it with model.apply(my_init).