search

ljleb / sd-webui-freeu

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

FreeU doesn't work when using with ControlNet #11

Closed MichaelWang1995 closed 1 year ago

MichaelWang1995 commented 1 year 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 1 year 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.