Skip to content

Commit 9563cee

Browse files
committed
Set eval on IPAdapter Flux
1 parent 233dffd commit 9563cee

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/diffusers/loaders/transformer_flux.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ def _convert_ip_adapter_image_proj_to_diffusers(self, state_dict, low_cpu_mem_us
7373
cross_attention_dim=cross_attention_dim,
7474
image_embed_dim=clip_embeddings_dim,
7575
num_image_text_embeds=num_image_text_embeds,
76-
)
76+
).eval()
7777

7878
for key, value in state_dict.items():
7979
diffusers_name = key.replace("proj", "image_embeds")
@@ -139,7 +139,7 @@ def _convert_ip_adapter_attn_to_diffusers(self, state_dicts, low_cpu_mem_usage=F
139139
num_tokens=num_image_text_embeds,
140140
dtype=self.dtype,
141141
device=self.device,
142-
)
142+
).eval()
143143

144144
value_dict = {}
145145
for i, state_dict in enumerate(state_dicts):
@@ -175,5 +175,5 @@ def _load_ip_adapter_weights(self, state_dicts, low_cpu_mem_usage=False):
175175
)
176176
image_projection_layers.append(image_projection_layer)
177177

178-
self.encoder_hid_proj = MultiIPAdapterImageProjection(image_projection_layers)
178+
self.encoder_hid_proj = MultiIPAdapterImageProjection(image_projection_layers).eval()
179179
self.config.encoder_hid_dim_type = "ip_image_proj"

0 commit comments

Comments
 (0)