mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-02-03 01:24:56 -05:00
Fix return type of prepare_noise_and_latents(...).
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user