mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-09-20 03:15:26 +00:00
fix broken test_ic_light
This commit is contained in:
parent
d174866ac0
commit
91913c76e6
|
@ -2611,7 +2611,7 @@ def test_ic_light(
|
||||||
negative_text="lowres, bad anatomy, bad hands, cropped, worst quality",
|
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"))
|
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)
|
x = torch.randn(1, 4, 64, 64, device=test_device)
|
||||||
for step in sd.steps:
|
for step in sd.steps:
|
||||||
x = sd(
|
x = sd(
|
||||||
|
|
Loading…
Reference in a new issue