Skip to content

Commit f86ccac

Browse files
committed
don't need the compilation test, we test it elsewhere.
1 parent 6cf2933 commit f86ccac

File tree

1 file changed

+0
-23
lines changed

1 file changed

+0
-23
lines changed

tests/lora/utils.py

Lines changed: 0 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@
1717
import os
1818
import re
1919
import tempfile
20-
import unittest
2120
from itertools import product
2221

2322
import numpy as np
@@ -1013,28 +1012,6 @@ def test_unexpected_keys_warning(self):
10131012

10141013
self.assertTrue(".diffusers_cat" in cap_logger.out)
10151014

1016-
@unittest.skip("This is failing for now - need to investigate")
1017-
def test_simple_inference_with_text_denoiser_lora_unfused_torch_compile(self):
1018-
"""
1019-
Tests a simple inference with lora attached to text encoder and unet, then unloads the lora weights
1020-
and makes sure it works as expected
1021-
"""
1022-
for scheduler_cls in self.scheduler_classes:
1023-
pipe, inputs, _, text_lora_config, denoiser_lora_config = self._setup_pipeline_and_get_base_output(
1024-
scheduler_cls
1025-
)
1026-
1027-
pipe, _ = self.add_adapters_to_pipeline(pipe, text_lora_config, denoiser_lora_config)
1028-
1029-
pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
1030-
pipe.text_encoder = torch.compile(pipe.text_encoder, mode="reduce-overhead", fullgraph=True)
1031-
1032-
if self.has_two_text_encoders or self.has_three_text_encoders:
1033-
pipe.text_encoder_2 = torch.compile(pipe.text_encoder_2, mode="reduce-overhead", fullgraph=True)
1034-
1035-
# Just makes sure it works..
1036-
_ = pipe(**inputs, generator=torch.manual_seed(0))[0]
1037-
10381015
def test_modify_padding_mode(self):
10391016
def set_pad_mode(network, mode="circular"):
10401017
for _, module in network.named_modules():

0 commit comments

Comments
 (0)