Skip to content

Commit b7e7ab3

Browse files
committed
update
1 parent 0c2e84e commit b7e7ab3

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

tests/lora/test_lora_layers_wan.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -147,10 +147,13 @@ def test_simple_inference_with_text_lora_fused(self):
147147
def test_simple_inference_with_text_lora_save_load(self):
148148
pass
149149

150-
# Refer to
151-
# https://github.com/huggingface/diffusers/pull/11806 for more details.
152150
@require_peft_version_greater("0.13.2")
153151
def test_lora_exclude_modules_for_wan(self):
152+
"""
153+
We test if the modules from `target_modules` and `exclude_modules`, that have
154+
overlap in their names, are impacted as expected. Refer to
155+
https://github.com/huggingface/diffusers/pull/11806 for more details.
156+
"""
154157
scheduler_cls = self.scheduler_classes[0]
155158
components, text_lora_config, denoiser_lora_config = self.get_dummy_components(scheduler_cls)
156159
pipe = self.pipeline_class(**components).to(torch_device)
@@ -184,5 +187,4 @@ def test_lora_exclude_modules_for_wan(self):
184187
check_module_lora_metadata(
185188
parsed_metadata=parsed_metadata, lora_metadatas=lora_metadatas, module_key="transformer"
186189
)
187-
188190
# Inference matching is already tested in `test_lora_exclude_modules`.

0 commit comments

Comments
 (0)