Skip to content

Commit 90fff36

Browse files
author
Attashe
committed
remove unused import
1 parent 86e46fb commit 90fff36

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

invokeai/app/invocations/flux_denoise.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -682,24 +682,24 @@ def _prep_uno_reference_imgs(self, context: InvocationContext) -> list[torch.Ten
682682
# Load the conditioning image and resize it to the target image size.
683683
assert self.controlnet_vae is not None, 'Controlnet Vae must be set for UNO encoding'
684684
vae_info = context.models.load(self.controlnet_vae.vae)
685-
685+
686686
assert self.uno_reference is not None, "Needs reference images for UNO"
687687

688688
ref_img_names: list[str] = self.uno_reference.image_names
689689
ref_latents: list[torch.Tensor] = []
690-
691-
# TODO: Maybe move reference side to UNO Node
690+
691+
# TODO: Maybe move reference side to UNO Node as parameter
692692
ref_long_side = 512 if len(ref_img_names) <= 1 else 320
693-
693+
694694
for img_name in ref_img_names:
695695
image_pil = context.images.get_pil(img_name)
696696
image_pil = image_pil.convert("RGB") # To correct resizing
697697
image_pil = preprocess_ref(image_pil, ref_long_side) # resize and crop
698-
698+
699699
image_tensor = (TVF.to_tensor(image_pil) * 2.0 - 1.0).unsqueeze(0).float()
700700
ref_latent = FluxVaeEncodeInvocation.vae_encode(vae_info=vae_info, image_tensor=image_tensor)
701701
ref_latents.append(ref_latent)
702-
702+
703703
return ref_latents
704704

705705
def _prep_structural_control_img_cond(self, context: InvocationContext) -> torch.Tensor | None:

invokeai/app/invocations/flux_uno.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from typing import Literal, Optional
1+
from typing import Optional
22

33
from PIL import Image
44

0 commit comments

Comments
 (0)