diff --git a/extensions-builtin/forge_preprocessor_clipvision/scripts/preprocessor_clipvision.py b/extensions-builtin/forge_preprocessor_clipvision/scripts/preprocessor_clipvision.py index 248ca124..8d971e01 100644 --- a/extensions-builtin/forge_preprocessor_clipvision/scripts/preprocessor_clipvision.py +++ b/extensions-builtin/forge_preprocessor_clipvision/scripts/preprocessor_clipvision.py @@ -27,7 +27,7 @@ class PreprocessorClipVision(Preprocessor): ) self.clipvision = ldm_patched.modules.clip_vision.load(ckpt_path) - input_image = numpy_to_pytorch(input_image).to(self.clipvision.patcher.current_device) + input_image = numpy_to_pytorch(input_image) return self.clipvision.encode_image(input_image) diff --git a/extensions-builtin/sd_forge_controlnet/lib_controlnet/controlnet_ui/controlnet_ui_group.py b/extensions-builtin/sd_forge_controlnet/lib_controlnet/controlnet_ui/controlnet_ui_group.py index 254bc86f..b19b8710 100644 --- a/extensions-builtin/sd_forge_controlnet/lib_controlnet/controlnet_ui/controlnet_ui_group.py +++ b/extensions-builtin/sd_forge_controlnet/lib_controlnet/controlnet_ui/controlnet_ui_group.py @@ -935,7 +935,9 @@ class ControlNetUiGroup(object): else None, ) - if not isinstance(result, np.ndarray) and result.nidm == 3 and result.shape[2] < 5: + is_image = isinstance(result, np.ndarray) and result.ndim == 3 and result.shape[2] < 5 + + if not is_image: result = img result = external_code.visualize_inpaint_mask(result)