IndexError: The shape of the mask [1, 1, 1, 512, 512] at index 2 does not match the shape of the indexed tensor [1, 1, 512, 512] at index 2

[hanyao666]

I tried running the code using the image in the sample, but encountered the following error. I am a beginner in this field, how can I solve it?
IndexError: The shape of the mask [1, 1, 1, 512, 512] at index 2 does not match the shape of the indexed tensor [1, 1, 512, 512] at index 2

[kosirobwada]

I encountered the same issue. Changing the line 288 in scripts/face_swap.py to fixed the problem.

hole_index = hole_map[None, None] == 255
is_foreground[hole_index] = True