AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-10-24 20:59:27 +03:00
Code Activity

[tests] fix lora tests involving clip. (#13675)

Browse Source 
fix lora tests involving clip.
This commit is contained in:
Sayak Paul
2026-05-06 08:23:14 +09:00
committed by GitHub
parent 2b9572d6ee
commit cff1e39f1b
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/lora/utils.py
View File

@@ -635,7 +635,7 @@ class PeftLoraLoaderMixinTests:
state_dict = {
f"text_encoder.{module_name}": param
for module_name, param in get_peft_model_state_dict(pipe.text_encoder).items()
if "text_model.encoder.layers.4" not in module_name
if "encoder.layers.4" not in module_name
}
if self.has_two_text_encoders or self.has_three_text_encoders:
@@ -644,7 +644,7 @@ class PeftLoraLoaderMixinTests:
{
f"text_encoder_2.{module_name}": param
for module_name, param in get_peft_model_state_dict(pipe.text_encoder_2).items()
if "text_model.encoder.layers.4" not in module_name
if "encoder.layers.4" not in module_name
}
)
@@ -776,8 +776,9 @@ class PeftLoraLoaderMixinTests:
)
if "text_encoder" in self.pipeline_class._lora_loadable_modules:
text_encoder_root = getattr(pipe.text_encoder, "text_model", pipe.text_encoder)
self.assertTrue(
pipe.text_encoder.text_model.encoder.layers[0].self_attn.q_proj.scaling["default"] == 1.0,
text_encoder_root.encoder.layers[0].self_attn.q_proj.scaling["default"] == 1.0,
"The scaling parameter has not been correctly restored!",
)