fix broken test_ic_light

Laurent · Laurent · commit d576311775e5 · 2024-08-21T08:34:28.000Z
diff --git a/tests/e2e/test_diffusion.py b/tests/e2e/test_diffusion.py
@@ -2611,7 +2611,7 @@ def test_ic_light(
         negative_text="lowres, bad anatomy, bad hands, cropped, worst quality",
     )
     ic_light_condition = sd.compute_gray_composite(image=kitchen_dog, mask=kitchen_dog_mask.convert("L"))
-    sd.set_ic_light_condition(ic_light_condition)
+    sd.set_ic_light_condition(ic_light_condition, use_rescaled_image=True)
     x = torch.randn(1, 4, 64, 64, device=test_device)
     for step in sd.steps:
         x = sd(

Original file line number	Diff line number	Diff line change
`@@ -2611,7 +2611,7 @@ def test_ic_light(`
`2611`	`2611`	`negative_text="lowres, bad anatomy, bad hands, cropped, worst quality",`
`2612`	`2612`	`)`
`2613`	`2613`	`ic_light_condition = sd.compute_gray_composite(image=kitchen_dog, mask=kitchen_dog_mask.convert("L"))`
`2614`		`- sd.set_ic_light_condition(ic_light_condition)`
	`2614`	`+ sd.set_ic_light_condition(ic_light_condition, use_rescaled_image=True)`
`2615`	`2615`	`x = torch.randn(1, 4, 64, 64, device=test_device)`
`2616`	`2616`	`for step in sd.steps:`
`2617`	`2617`	`x = sd(`