Revert "debug"

This reverts commit fb4aec0c.

Revert "debug"
This reverts commit fb4aec0c.
05b3d36a · sayakpaul · fb4aec0c · 05b3d36a
Commit 05b3d36a authored Jan 04, 2024 by sayakpaul
Hide 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]