Predict crashes if last batch is smaller then batch_size about tensorflow HOT 6 CLOSED

backbone = keras_cv.models.YOLOV8Backbone.from_preset(
                'yolo_v8_l_backbone',
                input_shape=[512,512, 3],
            )

model = keras_cv.models.YOLOV8Detector(
                    num_classes=1,
                    bounding_box_format='xyxy',
                    backbone=backbone,
                    fpn_depth=3,
                )

model.prediction_decoder = keras_cv.layers.NonMaxSuppression(
            bounding_box_format='xyxy',
            from_logits=True,
            iou_threshold=0.35,
            confidence_threshold=0.53,
            max_detections=50,
        )

def prepare_image(self: Self, input_image_path: Path) -> np.ndarray:
        try:
            input_image = cv2.imread(str(input_image_path))
            # input_image = cv2.bilateralFilter(input_image, 9, 75, 75)
            if self.color_mode == "rgb":
                input_image = cv2.cvtColor(input_image, cv2.COLOR_BGR2RGB)

            else:
                input_image = cv2.cvtColor(input_image, cv2.COLOR_BGR2GRAY)

            return input_image
tiles_paths = ['01.jpg', '02.jpg', ...]

image_patches = np.array(
  [
    prepare_image(input_image_path=tile_path)
                        for tile_path in tiles_paths
  ],
)
y_pred = model.predict(
   image_patches,
   batch_size=8,
)