Is there any way to inpaint with batch size > 1 with different mask for each image?

[davidboxboro]

I'd like to do inpainting inference on a dataset of images where the mask is different for each image. I'd like to use a batch size > 1 to increase inference speed. Is there any way to do this? If not, how can I perform inference more efficiently?