resolves issue #1

where swapped face outputs were not saved to the directory
xizzat · Jan 29, 2024 · a11c05d · a11c05d
1 parent dce7b6b
commit a11c05d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/modules/async_worker.py b/modules/async_worker.py
@@ -782,6 +782,10 @@ def callback(step, x0, x, total_steps, y):
                 if inpaint_worker.current_task is not None:
                     imgs = [inpaint_worker.current_task.post_process(x) for x in imgs]
 
+                if inswapper_enabled and input_image_checkbox:
+                    print("Lets swap some faces!")
+                    imgs = perform_face_swap(imgs, inswapper_source_image, inswapper_target_image_index)                    
+
                 for x in imgs:
                     d = [
                         ('Prompt', task['log_positive_prompt']),
@@ -809,10 +813,6 @@ def callback(step, x0, x, total_steps, y):
                     d.append(('Version', 'v' + fooocus_version.version))
                     log(x, d)
 
-                if inswapper_enabled and input_image_checkbox:
-                    print("Lets swap some faces!")
-                    imgs = perform_face_swap(imgs, inswapper_source_image, inswapper_target_image_index)
-
                 yield_result(async_task, imgs, do_not_show_finished_images=len(tasks) == 1)
             except ldm_patched.modules.model_management.InterruptProcessingException as e:
                 if shared.last_stop == 'skip':