Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-22 06:08:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

test_diffusion: remove debug leftovers

Browse source
This commit is contained in:
Cédric Deltheil 2023-09-29 18:46:42 +02:00 committed by Cédric Deltheil
parent 5fc6767a4a
commit 338042f332
1 changed files with 0 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/e2e/test_diffusion.py
View file

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