I think one check is sufficient: `if input_image.shape[:2] != inpaint_mask.shape`. I might even leave off the slicing since RGB images aren't supported yet.