FreeU support?

[asagi4]

Can AITemplate support the recent FreeU node added to ComfyUI, or at least its functionality? It uses a transformer patch which gets ignored by AIT inference. This is unfortunate since the FreeU algorithm can have very interesting effects on generation quality.

I tried poking around the code but I couldn't find a place to plug it in. It might be that it needs a specifically compiled FreeU module, but that goes over my head currently.

Is there a way for AIT to support FreeU? I'm not opposed to writing a patch myself, but I need a pointer in the right direction.