From 0e38928c8d45935697f3306a2e6d6f228d385eba Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?C=C3=A9dric=20Deltheil?= Date: Mon, 11 Sep 2023 17:02:35 +0200 Subject: [PATCH] sdxl test: add missing torch no_grad --- tests/e2e/test_diffusion.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/e2e/test_diffusion.py b/tests/e2e/test_diffusion.py index 2ac62a7..a9bfe96 100644 --- a/tests/e2e/test_diffusion.py +++ b/tests/e2e/test_diffusion.py @@ -1050,6 +1050,7 @@ def test_diffusion_ip_adapter( ensure_similar_images(predicted_image, expected_image_ip_adapter_woman) +@torch.no_grad() def test_sdxl_random_init( sdxl_ddim: StableDiffusion_XL, expected_sdxl_ddim_random_init: Image.Image, test_device: torch.device ) -> None: