1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

Update xformers SD3 test (#8712)

update
This commit is contained in:
Dhruv Nair
2024-06-27 01:54:27 +05:30
committed by GitHub
parent 5b51ad0052
commit effe4b9784
2 changed files with 8 additions and 0 deletions

View File

@@ -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():

View File

@@ -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