fix
This commit is contained in:
parent
179a753ce8
commit
459450c48c
@ -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)
|
||||
|
||||
|
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user