Revert "debug"

This reverts commit 7715e6c3.

Revert "debug"
This reverts commit 7715e6c3.
6dbef45e · sayakpaul · 7715e6c3 · 6dbef45e
Commit 6dbef45e authored Jan 04, 2024 by sayakpaul
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/lora/test_lora_layers_old_backend.py tests/lora/test_lora_layers_old_backend.py +1 -1

No files found.
--- 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).to(torch_device)
+            sd_pipe_loaded = StableDiffusionXLPipeline.from_pretrained(tmpdirname)

        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]