From cb4b6532950a1ded42f6fbc297aafa0f290cade7 Mon Sep 17 00:00:00 2001
From: SimoRyu <cloneofsimo@korea.ac.kr>
Date: Sat, 4 Feb 2023 15:31:48 +0000
Subject: [PATCH] much stricter mask

---
 lora_diffusion/dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lora_diffusion/dataset.py b/lora_diffusion/dataset.py
index eed4403..fa99ac1 100644
--- a/lora_diffusion/dataset.py
+++ b/lora_diffusion/dataset.py
@@ -267,7 +267,7 @@ def __getitem__(self, index):
                     Image.open(self.mask_path[index % self.num_instance_images])
                 )
                 * 0.5
-                + 1
+                + 0.5
             )
 
         if self.h_flip and random.random() > 0.5: