much stricter mask

cloneofsimo · Feb 4, 2023 · cb4b653 · cb4b653
1 parent cbd75b8
commit cb4b653
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: