Skip to content

GitLab

Commit fb4aec0c authored by sayakpaul's avatar sayakpaul
Browse files

debug

parent 63de23e3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/lora/test_lora_layers_old_backend.py
View file @ fb4aec0c
...@@ -1446,7 +1446,7 @@ class SDXLLoraLoaderMixinTests(unittest.TestCase): ...@@ -1446,7 +1446,7 @@ class SDXLLoraLoaderMixinTests(unittest.TestCase):
with tempfile.TemporaryDirectory() as tmpdirname: with tempfile.TemporaryDirectory() as tmpdirname:
sd_pipe.save_pretrained(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_images = sd_pipe_loaded(**pipeline_inputs, generator=torch.manual_seed(0)).images
loaded_lora_image_slice = loaded_lora_images[0, -3:, -3:, -1] loaded_lora_image_slice = loaded_lora_images[0, -3:, -3:, -1]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment