How to solve the problem of changing data sets,This happens in a different data set

[liuli996624]

(pytorch) D:\Python\Code\DF-DC\Face X-Ray\Face-X-Ray>python faceBlending.py
0it [02:34, ?it/s]
Traceback (most recent call last):
File "faceBlending.py", line 171, in
main()
File "faceBlending.py", line 74, in main
resultantFace = forge(srcFaceBgr, targetBgrT, blured) # forged face
File "faceBlending.py", line 111, in forge
return (mask * targetRgb + (1 - mask) * srcRgb).astype(np.uint8)
ValueError: operands could not be broadcast together with shapes (134,108,3) (128,106,3)