Skip to content

Commit 44e3847

Browse files
committed
Fix in loading state dict
1 parent 05f49e6 commit 44e3847

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/loaders/transformers_sd3.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,6 @@ def _load_ip_adapter_weights(self, state_dict: Dict, low_cpu_mem_usage: bool = _
8484
).to(device=self.device, dtype=self.dtype)
8585

8686
if not low_cpu_mem_usage:
87-
self.image_proj.load_state_dict(state_dict, strict=True)
87+
self.image_proj.load_state_dict(state_dict["image_proj"], strict=True)
8888
else:
89-
load_model_dict_into_meta(self.image_proj, state_dict, device=self.device, dtype=self.dtype)
89+
load_model_dict_into_meta(self.image_proj, state_dict["image_proj"], device=self.device, dtype=self.dtype)

0 commit comments

Comments
 (0)