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:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user