How to re-render input images generated from loading AI mat solution?

[Ryptun]

Question about these

These input images are being generated when I press "AI Mat Solution Load" which is nice. My problem is it doesn't always work correctly, it breaks on parented meshes, multimaterial objects can result in wrong normals. Button to just rebake those images without rebuilding whole graph would be nice.

What also would be nice would be an option to use custom Geometry node instead .Stack one

I'm currently making my own node to do similar thing, but more customizable and it would be really helpful if there was a field that would allow me to pick which geometry node I want to use for that mesh positioning.
You can see my prototype video here, it still has some work to do:

20250903200944_window_W9vOvPsszD.mp4