diff --git a/web_app/src/components/SidePanel/DiffusionOptions.tsx b/web_app/src/components/SidePanel/DiffusionOptions.tsx index 3048250..46d1059 100644 --- a/web_app/src/components/SidePanel/DiffusionOptions.tsx +++ b/web_app/src/components/SidePanel/DiffusionOptions.tsx @@ -62,6 +62,7 @@ const DiffusionOptions = () => { updateEnablePowerPaintV2, updateEnableBrushNet, updateEnableControlnet, + updateLCMLora, ] = useStore((state) => [ state.serverConfig.samplers, state.settings, @@ -77,6 +78,7 @@ const DiffusionOptions = () => { state.updateEnablePowerPaintV2, state.updateEnableBrushNet, state.updateEnableControlnet, + state.updateLCMLora, ]) const [exampleImage, isExampleImageLoaded] = useImage(paintByExampleFile) const negativePromptRef = useRef(null) @@ -295,7 +297,7 @@ const DiffusionOptions = () => { id="lcm-lora" checked={settings.enableLCMLora} onCheckedChange={(value) => { - updateSettings({ enableLCMLora: value }) + updateLCMLora(value) }} /> diff --git a/web_app/src/lib/states.ts b/web_app/src/lib/states.ts index 0a67348..53b4d74 100644 --- a/web_app/src/lib/states.ts +++ b/web_app/src/lib/states.ts @@ -200,6 +200,7 @@ type AppAction = { updateEnablePowerPaintV2: (newValue: boolean) => void updateEnableBrushNet: (newValue: boolean) => void updateEnableControlnet: (newValue: boolean) => void + updateLCMLora: (newValue: boolean) => void setModel: (newModel: ModelInfo) => void updateFileManagerState: (newState: Partial) => void @@ -840,6 +841,16 @@ export const useStore = createWithEqualityFn()( } }, + updateLCMLora(newValue) { + get().updateSettings({ enableLCMLora: newValue }) + if (newValue) { + get().updateSettings({ + enablePowerPaintV2: false, + enableBrushNet: false, + }) + } + }, + setModel: (newModel: ModelInfo) => { set((state) => { state.settings.model = newModel