Skip to content

Commit 71cc201

Browse files
authored
Fix FluxIPAdapterTesterMixin (#10354)
1 parent c34fc34 commit 71cc201

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/diffusers/loaders/transformer_flux.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -177,3 +177,5 @@ def _load_ip_adapter_weights(self, state_dicts, low_cpu_mem_usage=False):
177177

178178
self.encoder_hid_proj = MultiIPAdapterImageProjection(image_projection_layers)
179179
self.config.encoder_hid_dim_type = "ip_image_proj"
180+
181+
self.to(dtype=self.dtype, device=self.device)

0 commit comments

Comments
 (0)