From 929ee203c24e700ae744ac5534c90531f71ccaca Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Mon, 29 Jan 2024 15:43:43 -0800 Subject: [PATCH] Update controlnet.py --- extensions-builtin/sd_forge_controlnet/scripts/controlnet.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py index 97bcd8d8..67678c69 100644 --- a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py +++ b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py @@ -855,6 +855,8 @@ class ControlNetForForgeOfficial(scripts.Script): def process_unit_after_click_generate(self, p, unit, params, *args, **kwargs): h, w, hr_y, hr_x = self.get_target_dimensions(p) input_image, resize_mode = self.choose_input_image(p, unit) + assert isinstance(input_image, np.ndarray), 'Invalid input image!' + return def process_unit_before_every_sampling(self, p, unit, params, *args, **kwargs):