search

ljleb / sd-webui-freeu

a1111 implementation of https://github.com/ChenyangSi/FreeU
MIT License
301 stars 16 forks source link

FreeU doesn't work when using with ControlNet #11

Closed MichaelWang1995 closed 11 months ago

MichaelWang1995 commented 11 months ago

When using with Controlnet (any control type), FreeU will have not effect. Control net ver: 1.1.410 No error or message in command prompt.

Works fine with all other extensions I have.

ljleb commented 11 months ago

Controlnet overrides the unet forward function, we need to patch their unet as well. Having all these unet functions to copy paste is no good, we need to find a solution to only patch the torch.cat in the decoder block for-loop.