This commit is contained in:
Vladimir Mandic
2023-09-11 09:55:28 -04:00
parent 1f730b129f
commit 2f071c6586
4 changed files with 2 additions and 3 deletions
-1
View File
@@ -212,7 +212,6 @@ class StableDiffusionProcessing:
conditioning_mask = np.array(image_mask.convert("L"))
conditioning_mask = conditioning_mask.astype(np.float32) / 255.0
conditioning_mask = torch.from_numpy(conditioning_mask[None, None])
# Inpainting model uses a discretized mask as input, so we round to either 1.0 or 0.0
conditioning_mask = torch.round(conditioning_mask)
else: