Upload folder using huggingface_hub

gray-inpaint/config.json

@@ -10,6 +10,6 @@
   "height": 512,
   "model_type": "sd_gray_inpaint",
   "torch_dtype": "float32",
-  "transformers_version": "4.46.3",
+  "transformers_version": "4.47.0",
   "width": 512
 }

gray-inpaint/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2c6d964dca7f33a3a87e90056e8ab617efeabd99e3cfcea71f73d459b133f231
+oid sha256:9d7e04ec5bbbda567c7fa7e1d8478f997ec5bc1376dc3fa633dc737171b774da
 size 4055354432

gray-inpaint/modeling_sd_gray_inpaint.py

@@ -64,6 +64,7 @@ class SDGrayInpaintModel(PreTrainedModel):
             masks_logits = self.mask_predictor(images_gray_masked)
             masks = (torch.sigmoid(masks_logits)>0.5)*1.
         masks = masks.float().to(self.vae.device)
+        images_gray_masked = (1-masks) * images_gray_masked
         B, C, H, W = images_gray_masked.shape
         prompt_embeds = self.prompt_embeds.repeat(B,1,1)
 

modeling_sd_gray_inpaint.py

@@ -64,6 +64,7 @@ class SDGrayInpaintModel(PreTrainedModel):
             masks_logits = self.mask_predictor(images_gray_masked)
             masks = (torch.sigmoid(masks_logits)>0.5)*1.
         masks = masks.float().to(self.vae.device)
+        images_gray_masked = (1-masks) * images_gray_masked
         B, C, H, W = images_gray_masked.shape
         prompt_embeds = self.prompt_embeds.repeat(B,1,1)