diff --git a/src/diffusers/loaders/lora_pipeline.py b/src/diffusers/loaders/lora_pipeline.py index 2c29702490..7c92bf0761 100644 --- a/src/diffusers/loaders/lora_pipeline.py +++ b/src/diffusers/loaders/lora_pipeline.py @@ -4810,8 +4810,8 @@ class WanLoraLoaderMixin(LoraBaseMixin): transformer: torch.nn.Module, state_dict, ): - if transformer.config.image_dim is None: - return state_dict + # if transformer.config.image_dim is None: + # return state_dict target_device = transformer.device