Allow empty inpainting controlnet init image

AbdullahAlfaraj · Oct 27, 2023 · 91e0830 · 91e0830
1 parent 321d448
commit 91e0830
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/typescripts/session/modes.ts b/typescripts/session/modes.ts
@@ -161,8 +161,15 @@ export class Txt2ImgMode extends Mode {
             }
             control_networks[index] = true
 
+            const is_inpaint_model: boolean =
+                control_net_settings['controlnet_units'][index][
+                    'module'
+                ].includes('inpaint')
             if (
-                !control_net_settings['controlnet_units'][index]['input_image']
+                !control_net_settings['controlnet_units'][index][
+                    'input_image'
+                ] &&
+                !is_inpaint_model
             ) {
                 //@ts-ignore
                 app.showAlert('you need to add a valid ControlNet input image')
@@ -304,8 +311,17 @@ export class Img2ImgMode extends Mode {
                 continue
             }
             control_networks[index] = true
+
+            const is_inpaint_model: boolean =
+                control_net_settings['controlnet_units'][index][
+                    'module'
+                ].includes('inpaint')
+
             if (
-                !control_net_settings['controlnet_units'][index]['input_image']
+                !control_net_settings['controlnet_units'][index][
+                    'input_image'
+                ] &&
+                !is_inpaint_model
             ) {
                 //@ts-ignore
                 app.showAlert('you need to add a valid ControlNet input image')