在ComfyUI中,将图片通过controlnet后送入sampler,流程能正常运行。有些应用中为了去除图片背景的污染,采用RMBG节点将背景去除后再输入时出现类似以下的报错:
RuntimeError: Given groups=1, weight of size [16, 3, 3, 3], expected input[1, 4, 1344, 768] to have 3 channels, but got 4 channels instead
原因为张量输入数据的张量通过不匹配。采用debug获得并比较数据两者数据的差异,如下:
ttN debug:
tensor([[[[0.0000, 0.0000, 0.0000],
[0.0000, 0.0000, 0.0000],
[0.0000, 0.0000, 0.0000],
...,
[0.0000, 0.0039, 0.0039],
[0.0000, 0.0039, 0.0039],
[0.0039, 0.0039, 0.0039]],
[[0.0000, 0.0000, 0.0000],
[0.0000, 0.0000, 0.0000],
[0.0000, 0.0000, 0.0000],
...,
&nbs