File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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`.
You can’t perform that action at this time.
0 commit comments