Fixing the equation on the "slice_w" value

The equation was wrong; it should be simply the opposite.

This will help to crop image with the right sizes. Solves the problem mentioned here: https://github.com/ali-vilab/ACE_plus/issues/90
This commit is contained in:
Astroboy
2025-04-06 01:30:47 +03:00
committed by GitHub
parent ac1654dba3
commit abaf9dad6a
@@ -391,7 +391,7 @@ class AcePlusFFTProcessor:
edit_image = edit_image * (1 - edit_mask)
edit_image = edit_image.unsqueeze(0).permute(0, 2, 3, 1)
change_image = change_image.unsqueeze(0).permute(0, 2, 3, 1)
slice_w = slice_w if slice_w < 30 else slice_w + 30
slice_w = slice_w if slice_w > 30 else slice_w + 30
return edit_image + 0.5, change_image + 0.5, edit_mask, out_h, out_w, slice_w