optional progress callback

pipeline.py

@@ -1,4 +1,5 @@
 from dataclasses import dataclass
+from typing import Callable, Optional
 
 import cv2
 import numpy as np
@@ -97,7 +98,12 @@ class InstantMeshPipeline(DiffusionPipeline):
         self.register_modules(lrm=self.lrm)
 
     @torch.no_grad()
-    def __call__(self, images: torch.Tensor, generate_texture: bool = False):
+    def __call__(
+        self,
+        images: torch.Tensor,
+        generate_texture: bool = False,
+        progress_callback: Optional[Callable[[float], None]] = None,
+    ):
         self.lrm.init_flexicubes_geometry(self._execution_device, fovy=30.0)
         cameras = get_zero123plus_input_cameras().to(self._execution_device)
         planes = self.lrm.forward_planes(images, cameras)
@@ -107,6 +113,7 @@ class InstantMeshPipeline(DiffusionPipeline):
                 planes,
                 use_texture_map=True,
                 texture_resolution=1024,
+                progress_callback=progress_callback,
             )
             vertices, vertex_indices, uvs, uv_indices, texture = mesh_out
 
@@ -149,6 +156,7 @@ class InstantMeshPipeline(DiffusionPipeline):
             mesh_out = self.lrm.extract_mesh(
                 planes,
                 use_texture_map=False,
+                progress_callback=progress_callback,
             )
             vertices, faces, _ = mesh_out