Fix return type of prepare_noise_and_latents(...).

This commit is contained in:
Ryan Dick 2024-06-07 11:01:50 -04:00 committed by Kent Keirsey
parent 3d49541c09
commit 020e8eb413

View File

@ -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)