mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 23:12:02 +00:00
test_diffusion: remove debug leftovers
This commit is contained in:
parent
5fc6767a4a
commit
338042f332
|
@ -1269,7 +1269,6 @@ def test_diffusion_ip_adapter_plus(
|
||||||
condition_scale=7.5,
|
condition_scale=7.5,
|
||||||
)
|
)
|
||||||
predicted_image = sd15.lda.decode_latents(x)
|
predicted_image = sd15.lda.decode_latents(x)
|
||||||
predicted_image.save("output.png")
|
|
||||||
|
|
||||||
ensure_similar_images(predicted_image, expected_image_ip_adapter_plus_statue, min_psnr=35, min_ssim=0.98)
|
ensure_similar_images(predicted_image, expected_image_ip_adapter_plus_statue, min_psnr=35, min_ssim=0.98)
|
||||||
|
|
||||||
|
@ -1326,7 +1325,6 @@ def test_diffusion_sdxl_ip_adapter_plus(
|
||||||
)
|
)
|
||||||
sdxl.lda.to(dtype=torch.float32)
|
sdxl.lda.to(dtype=torch.float32)
|
||||||
predicted_image = sdxl.lda.decode_latents(x.to(dtype=torch.float32))
|
predicted_image = sdxl.lda.decode_latents(x.to(dtype=torch.float32))
|
||||||
predicted_image.save("output.png")
|
|
||||||
|
|
||||||
ensure_similar_images(predicted_image, expected_image_sdxl_ip_adapter_plus_woman)
|
ensure_similar_images(predicted_image, expected_image_sdxl_ip_adapter_plus_woman)
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue