Update ink_detection_pipeline.py

fixing bfloat16 handling on certain devices

ink_detection_pipeline.py

@@ -72,7 +72,7 @@ class InkDetectionPipeline(Pipeline):
             sub_y_preds = torch.sigmoid(sub_y_preds)
 
             # Move to CPU and numpy
-            sub_y_preds = sub_y_preds.detach().cpu().numpy()
+            sub_y_preds = sub_y_preds.detach().cpu().float().numpy()
             # shape (subB, 1, tile_size, tile_size)
 
             all_preds.append(sub_y_preds)