From fb4aec0ce37f418113b2f20138fef8bd8484af25 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Thu, 4 Jan 2024 10:37:28 +0530 Subject: [PATCH] debug --- tests/lora/test_lora_layers_old_backend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/lora/test_lora_layers_old_backend.py b/tests/lora/test_lora_layers_old_backend.py index 09bb87c851..d5066c0530 100644 --- a/tests/lora/test_lora_layers_old_backend.py +++ b/tests/lora/test_lora_layers_old_backend.py @@ -1446,7 +1446,7 @@ class SDXLLoraLoaderMixinTests(unittest.TestCase): with tempfile.TemporaryDirectory() as tmpdirname: sd_pipe.save_pretrained(tmpdirname) - sd_pipe_loaded = StableDiffusionXLPipeline.from_pretrained(tmpdirname) + sd_pipe_loaded = StableDiffusionXLPipeline.from_pretrained(tmpdirname).to(torch_device) loaded_lora_images = sd_pipe_loaded(**pipeline_inputs, generator=torch.manual_seed(0)).images loaded_lora_image_slice = loaded_lora_images[0, -3:, -3:, -1]