fix unit test bug

open-mmlab · Aug 9, 2023 · bc9f677 · bc9f677
1 parent 1334067
commit bc9f677
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_models/test_editors/test_stable_diffusion/test_stable_diffusion_inpaint.py b/tests/test_models/test_editors/test_stable_diffusion/test_stable_diffusion_inpaint.py
@@ -102,7 +102,7 @@ def test_stable_diffusion():
     StableDiffuser.text_encoder = dummy_text_encoder()
 
     image = torch.clip(torch.randn((1, 3, 64, 64)), -1, 1)
-    mask = torch.clip(torch.randn((1, 3, 64, 64)), -1, 1)
+    mask = torch.clip(torch.randn((1, 3, 64, 64)), 0, 1)
 
     with pytest.raises(Exception):
         StableDiffuser.infer('temp', image, mask, height=31, width=31)