Extension isn't working right in SD Forge Neo fork (https://github.com/Haoming02/sd-webui-forge-classic/tree/neo)