github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-12 09:34:54 -05:00
Code Issues Packages Projects Releases Wiki Activity

tidy(nodes): remove extraneous comments & add useful ones

Browse Source
This commit is contained in:
psychedelicious
2025-06-27 17:41:27 +10:00
parent 075345bffd
commit 9066dc1839
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
invokeai/app/invocations/flux_denoise.py
View File

@@ -147,6 +147,9 @@ class FluxDenoiseInvocation(BaseInvocation, WithMetadata, WithBoard):
description=FieldDescriptions.vae,
input=Input.Connection,
)
# This node accepts a images for features like FLUX Fill, ControlNet, and Kontext, but needs to operate on them in
# latent space. We'll run the VAE to encode them in this node instead of requiring the user to run the VAE in
# upstream nodes.
ip_adapter: IPAdapterField | list[IPAdapterField] | None = InputField(
description=FieldDescriptions.ip_adapter, title="IP-Adapter", default=None, input=Input.Connection
@@ -386,29 +389,26 @@ class FluxDenoiseInvocation(BaseInvocation, WithMetadata, WithBoard):
kontext_extension = None
if self.kontext_conditioning is not None:
# We need a VAE to encode the reference image. We can reuse the
# controlnet_vae field as it serves a similar purpose (image to latents).
if not self.controlnet_vae:
raise ValueError("A VAE (e.g., controlnet_vae) must be provided to use Kontext conditioning.")
kontext_extension = KontextExtension(
kontext_field=self.kontext_conditioning,
context=context,
vae_field=self.controlnet_vae, # Pass the VAE field
vae_field=self.controlnet_vae,
device=TorchDevice.choose_torch_device(),
dtype=inference_dtype,
)
final_img, final_img_ids = x, img_ids
original_seq_len = x.shape[1] # Store the original sequence length
original_seq_len = x.shape[1]
if kontext_extension is not None:
final_img, final_img_ids = kontext_extension.apply(final_img, final_img_ids)
# The denoise function will now use the combined tensors
x = denoise(
model=transformer,
img=final_img, # Pass the combined image tokens
img_ids=final_img_ids, # Pass the combined image IDs
img=final_img,
img_ids=final_img_ids,
pos_regional_prompting_extension=pos_regional_prompting_extension,
neg_regional_prompting_extension=neg_regional_prompting_extension,
timesteps=timesteps,