Recrosoftware/sd-webui-reactor
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

save all images in extras tab

Browse Source
This commit is contained in:
Andray 2024-03-21 13:28:01 +04:00
parent ae57149962
commit b7b094ba78
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/reactor_faceswap.py
View File

@ -500,7 +500,7 @@ class FaceSwapScriptExtras(scripts_postprocessing.ScriptPostprocessing):
# TAB MAIN
msgs: dict = {
"extra_multiple_source": " | Сomparison grid as a result",
"extra_multiple_source": "",
}
img, imgs, select_source, face_model, source_folder, save_original, mask_face, source_faces_index, gender_source, faces_index, gender_target, face_restorer_name, face_restorer_visibility, codeformer_weight, swap_in_source, swap_in_generated, random_image = ui_main.show(is_img2img=False, show_br=False, **msgs)
@ -684,6 +684,10 @@ class FaceSwapScriptExtras(scripts_postprocessing.ScriptPostprocessing):
if len(result) > 0 and swapped > 0:
image = result[0]
if len(result) > 1:
if hasattr(pp, 'extra_images'):
image = result[0]
pp.extra_images.extend(result[1:])
else:
grid = make_grid(result)
result.insert(0, grid)
image = grid