Compare commits

..

No commits in common. "bfd79956ab295cbbe07ede33048dc189363e3435" and "a660ebcd1b8da02ca5d6313df9802c6d0112f79a" have entirely different histories.

View File

@ -86,6 +86,6 @@ def image_alpha_fix(destination, source):
if destination.shape[-1] < source.shape[-1]:
source = source[...,:destination.shape[-1]]
elif destination.shape[-1] > source.shape[-1]:
source = torch.nn.functional.pad(source, (0, 1))
source[..., -1] = 1.0
destination = torch.nn.functional.pad(destination, (0, 1))
destination[..., -1] = 1.0
return destination, source