From 020e8eb41319dce2d3abfa137d5131e27bef07d6 Mon Sep 17 00:00:00 2001 From: Ryan Dick Date: Fri, 7 Jun 2024 11:01:50 -0400 Subject: [PATCH] Fix return type of prepare_noise_and_latents(...). --- invokeai/app/invocations/denoise_latents.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/invokeai/app/invocations/denoise_latents.py b/invokeai/app/invocations/denoise_latents.py index 26d826ab72..0f4fcd1ef0 100644 --- a/invokeai/app/invocations/denoise_latents.py +++ b/invokeai/app/invocations/denoise_latents.py @@ -668,7 +668,7 @@ class DenoiseLatentsInvocation(BaseInvocation): @staticmethod def prepare_noise_and_latents( context: InvocationContext, noise_field: LatentsField | None, latents_field: LatentsField | None - ) -> Tuple[float, torch.Tensor | None, torch.Tensor]: + ) -> Tuple[int, torch.Tensor | None, torch.Tensor]: noise = None if noise_field is not None: noise = context.tensors.load(noise_field.latents_name)