AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

vace

Browse Source
This commit is contained in:
linoytsaban
2025-05-27 20:51:53 +03:00
parent b7b1abdf13
commit 3ac5a286a4
1 changed files with 1 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/diffusers/loaders/lora_pipeline.py
View File

@@ -4810,7 +4810,6 @@ class WanLoraLoaderMixin(LoraBaseMixin):
transformer: torch.nn.Module,
state_dict,
):
print("BEFORE", list(state_dict.keys()))
if transformer.config.image_dim is None:
return state_dict
@@ -4912,10 +4911,6 @@ class WanLoraLoaderMixin(LoraBaseMixin):
print("AFTER 2:", list(state_dict.keys()))
return state_dict
return state_dict
def load_lora_weights(
self,
pretrained_model_name_or_path_or_dict: Union[str, Dict[str, torch.Tensor]],
@@ -4972,7 +4967,7 @@ class WanLoraLoaderMixin(LoraBaseMixin):
if not is_correct_format:
raise ValueError("Invalid LoRA checkpoint.")
print("AFTER:", list(state_dict.keys()))
print("WTF")
self.load_lora_into_transformer(
state_dict,
transformer=getattr(self, self.transformer_name) if not hasattr(self, "transformer") else self.transformer,