Fix FluxIPAdapterTesterMixin (#10354)

huggingface · Dec 23, 2024 · 71cc201 · 71cc201
1 parent c34fc34
commit 71cc201
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/diffusers/loaders/transformer_flux.py b/src/diffusers/loaders/transformer_flux.py
@@ -177,3 +177,5 @@ def _load_ip_adapter_weights(self, state_dicts, low_cpu_mem_usage=False):
 
         self.encoder_hid_proj = MultiIPAdapterImageProjection(image_projection_layers)
         self.config.encoder_hid_dim_type = "ip_image_proj"
+
+        self.to(dtype=self.dtype, device=self.device)