How do I integrate a custom U-Net into webui pipeline?

[nam-drun]

So my thesis is using Neural Architecture Search to find better Diffusion U-Net. I want to integrate it to webui just to test how would custom Diffusion U-Net would roll with existing tools on webui

[78Alpha]

https://github.com/AUTOMATIC1111/stable-diffusion-webui-tensorrt

You can checkout the official tensorrt implementation to see how a new unet is implemented. Depending on how different it is to the normal ckpt and safetensor formats, it usually means dynamic shape issues, controlnet not working, loras not working, etc.