From 71cc2013fe9a1cf3bbd9fdcdff5dbf7b2f8d9ee9 Mon Sep 17 00:00:00 2001 From: hlky Date: Mon, 23 Dec 2024 08:50:06 +0000 Subject: [PATCH] Fix FluxIPAdapterTesterMixin (#10354) --- src/diffusers/loaders/transformer_flux.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/diffusers/loaders/transformer_flux.py b/src/diffusers/loaders/transformer_flux.py index 52a48e56e748..9fe712bb12e9 100644 --- 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)