Merge pull request #3 from yoinked-h/patch-1

add noise mask support (inpainting)
2023-11-27 01:26:14 +01:00
parent 82b7b817a3 a04a7cacd7
commit 89543b07f2
1 changed files with 4 additions and 0 deletions
@@ -90,6 +90,10 @@ class LatentUpscaler:
 		lt = samples["samples"]
 		lt = model(lt)
 		del model
+		if "noise_mask" in samples.keys():
+			# expand the noise mask to the same shape as the latent
+			mask = torch.nn.functional.interpolate(samples['noise_mask'], scale_factor=float(scale_factor), mode='bicubic') 
+			return ({"samples": lt, "noise_mask": mask},)
 		return ({"samples": lt},)

 NODE_CLASS_MAPPINGS = {