From 40c33b9595326f339effaf5c7b29480700358bea Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?C=C3=A9dric=20Deltheil?= Date: Sun, 21 Jan 2024 18:58:44 +0100 Subject: [PATCH] rollback to 50 inference steps in IP-Adapter tests Follow up of 8a36c8c This is what is used in the official notebook (ip_adapter_demo.ipynb and ip_adapter-plus_demo.ipynb) --- tests/e2e/test_diffusion.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/e2e/test_diffusion.py b/tests/e2e/test_diffusion.py index bdfb048..0c544e0 100644 --- a/tests/e2e/test_diffusion.py +++ b/tests/e2e/test_diffusion.py @@ -1236,7 +1236,7 @@ def test_diffusion_ip_adapter( clip_image_embedding = ip_adapter.compute_clip_image_embedding(ip_adapter.preprocess_image(woman_image)) ip_adapter.set_clip_image_embedding(clip_image_embedding) - sd15.set_inference_steps(30) + sd15.set_inference_steps(50) manual_seed(2) x = torch.randn(1, 4, 64, 64, device=test_device, dtype=torch.float16) @@ -1382,7 +1382,7 @@ def test_diffusion_ip_adapter_plus( clip_image_embedding = ip_adapter.compute_clip_image_embedding(ip_adapter.preprocess_image(statue_image)) ip_adapter.set_clip_image_embedding(clip_image_embedding) - sd15.set_inference_steps(30) + sd15.set_inference_steps(50) manual_seed(42) # seed=42 is used in the official IP-Adapter demo x = torch.randn(1, 4, 64, 64, device=test_device, dtype=torch.float16)