it works great if the batch size is one, but it will break if the batch size is more t

<a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/us

comfyUi run more than 1 batch size at a time about abg-comfyui HOT 3 OPEN

kwaroran commented on September 23, 2024

comfyUi run more than 1 batch size at a time

from abg-comfyui.

Comments (3)

rzezagar commented on September 23, 2024

@arroyoDev Thanks for raising the issue. Have you found an alternative to work with animediff?

from abg-comfyui.

retrouve commented on September 23, 2024

Go find abg-comfyui folder in customNode, Then change RemoveImageBackgroundabg class in the init.py to

class RemoveImageBackgroundabg:
    def __init__(self):
        pass
    
    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "images": ("IMAGE",),
            },
        }

    RETURN_TYPES = ("IMAGE",)
    FUNCTION = "abg_remover"
    CATEGORY = "image"

    def abg_remover(self, images):
        batch_tensor = []
        for image in images:
            npa = image2nparray(image)
            print(npa.ndim)
            rmb = rmbg_fn(npa)
            image = nparray2image(rmb)
            batch_tensor.append(image)
        
        batch_tensor = torch.cat(batch_tensor, dim=0)
        return (batch_tensor,)

Then you will be fine.

from abg-comfyui.