Revert "debug"

This reverts commit fb4aec0.
absalan · Jan 4, 2024 · 05b3d36 · 05b3d36
1 parent fb4aec0
commit 05b3d36
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/lora/test_lora_layers_old_backend.py b/tests/lora/test_lora_layers_old_backend.py
@@ -1446,7 +1446,7 @@ def test_save_load_fused_lora_modules(self):
 
         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]