Merge pull request #223 from Haidra-Org/main

fix: callback during each step of post-proc.
Haidra-Org · Mar 21, 2024 · db81681 · db81681
2 parents a67579c + f0e2fc1
commit db81681
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/hordelib/horde.py b/hordelib/horde.py
@@ -1127,6 +1127,17 @@ def _default_progress_callback(comfyui_progress: ComfyUIProgress, message: str)
                 final_image = ret.image
                 final_rawpng = ret.rawpng
 
+                if progress_callback is not None:
+                    try:
+                        progress_callback(
+                            ProgressReport(
+                                hordelib_progress_state=ProgressState.progress,
+                                hordelib_message="Post Processing new image.",
+                            ),
+                        )
+                    except Exception as e:
+                        logger.error(f"Progress callback failed ({type(e)}): {e}")
+
                 # Ensure facefixers always happen first
                 post_processing_requested = sorted(
                     post_processing_requested,