This commit is contained in:
lllyasviel
2024-01-27 09:44:44 -08:00
parent b91ada8df3
commit aec5053938
6 changed files with 34 additions and 5 deletions

View File

@@ -35,6 +35,7 @@ class Blend:
CATEGORY = "image/postprocessing"
def blend_images(self, image1: torch.Tensor, image2: torch.Tensor, blend_factor: float, blend_mode: str):
image2 = image2.to(image1.device)
if image1.shape != image2.shape:
image2 = image2.permute(0, 3, 1, 2)
image2 = ldm_patched.modules.utils.common_upscale(image2, image1.shape[2], image1.shape[1], upscale_method='bicubic', crop='center')