sdxl test: add missing torch no_grad

This commit is contained in:
Cédric Deltheil 2023-09-11 17:02:35 +02:00 committed by Cédric Deltheil
parent b515c02867
commit 0e38928c8d

View file

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