fix: detailer exception when noise_mask is false

2023-10-06 23:22:54 +09:00
parent 6391fcb457
commit ebfe5bb3e3
2 changed files with 2 additions and 2 deletions
@@ -2,7 +2,7 @@ import configparser
 import os


-version = "V4.16"
+version = "V4.16.1"

 dependency_version = 13

@@ -163,7 +163,7 @@ def enhance_detail(image, model, clip, vae, guide_size, guide_size_for_bbox, max
                   detailer_hook=None,
                   refiner_ratio=None, refiner_model=None, refiner_clip=None, refiner_positive=None,
                   refiner_negative=None, control_net_wrapper=None):
-    if len(noise_mask.shape) == 3:
+    if noise_mask is not None and len(noise_mask.shape) == 3:
        noise_mask = noise_mask.squeeze(0)

    if wildcard_opt is not None and wildcard_opt != "":