From f91dae9be30b7c1e6caede9d32dcc2c7e955b340 Mon Sep 17 00:00:00 2001
From: linoytsaban <linoy@huggingface.co>
Date: Wed, 21 May 2025 19:13:21 +0300
Subject: [PATCH] revert

---
 src/diffusers/loaders/lora_pipeline.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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