mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-27 17:22:53 +03:00
@@ -382,6 +382,10 @@ class ModelTesterMixin:
|
||||
# If not has `set_attn_processor`, skip test
|
||||
return
|
||||
|
||||
if not hasattr(model, "set_default_attn_processor"):
|
||||
# If not has `set_attn_processor`, skip test
|
||||
return
|
||||
|
||||
model.set_default_attn_processor()
|
||||
assert all(type(proc) == AttnProcessor for proc in model.attn_processors.values())
|
||||
with torch.no_grad():
|
||||
|
||||
@@ -190,6 +190,10 @@ class StableDiffusion3ControlNetPipelineFastTests(unittest.TestCase, PipelineTes
|
||||
np.abs(image_slice.flatten() - expected_slice).max() < 1e-2
|
||||
), f"Expected: {expected_slice}, got: {image_slice.flatten()}"
|
||||
|
||||
@unittest.skip("xFormersAttnProcessor does not work with SD3 Joint Attention")
|
||||
def test_xformers_attention_forwardGenerator_pass(self):
|
||||
pass
|
||||
|
||||
|
||||
@slow
|
||||
@require_torch_gpu
|
||||
|
||||
Reference in New Issue
Block a user