fix broken test_ic_light

This commit is contained in:
Laurent 2024-08-20 16:46:55 +00:00
parent d174866ac0
commit 91913c76e6
No known key found for this signature in database

View file

@ -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(