Fix problem.

This commit is contained in:
comfyanonymous 2025-04-03 20:57:59 -04:00
parent 3d2e3a6f29
commit 721253cb05

View File

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