From bc5f356390c70372cc226ba097c4831c0ae77993 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Thu, 8 Feb 2024 07:55:36 +1100
Subject: [PATCH] feat(nodes): use LATENT_SCALE_FACTOR const in tensor output
 builders

---
 invokeai/app/invocations/noise.py      | 5 +++--
 invokeai/app/invocations/primitives.py | 5 +++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/invokeai/app/invocations/noise.py b/invokeai/app/invocations/noise.py
index 74b3d6e4cb..4093030388 100644
--- a/invokeai/app/invocations/noise.py
+++ b/invokeai/app/invocations/noise.py
@@ -5,6 +5,7 @@ import torch
 from pydantic import field_validator
 
 from invokeai.app.invocations.fields import FieldDescriptions, InputField, LatentsField, OutputField
+from invokeai.app.invocations.latent import LATENT_SCALE_FACTOR
 from invokeai.app.services.shared.invocation_context import InvocationContext
 from invokeai.app.util.misc import SEED_MAX
 
@@ -70,8 +71,8 @@ class NoiseOutput(BaseInvocationOutput):
     def build(cls, latents_name: str, latents: torch.Tensor, seed: int) -> "NoiseOutput":
         return cls(
             noise=LatentsField(latents_name=latents_name, seed=seed),
-            width=latents.size()[3] * 8,
-            height=latents.size()[2] * 8,
+            width=latents.size()[3] * LATENT_SCALE_FACTOR,
+            height=latents.size()[2] * LATENT_SCALE_FACTOR,
         )
 
 
diff --git a/invokeai/app/invocations/primitives.py b/invokeai/app/invocations/primitives.py
index d0f95c92d0..2a9cb8cf9b 100644
--- a/invokeai/app/invocations/primitives.py
+++ b/invokeai/app/invocations/primitives.py
@@ -16,6 +16,7 @@ from invokeai.app.invocations.fields import (
     OutputField,
     UIComponent,
 )
+from invokeai.app.invocations.latent import LATENT_SCALE_FACTOR
 from invokeai.app.services.images.images_common import ImageDTO
 from invokeai.app.services.shared.invocation_context import InvocationContext
 
@@ -321,8 +322,8 @@ class LatentsOutput(BaseInvocationOutput):
     def build(cls, latents_name: str, latents: torch.Tensor, seed: Optional[int] = None) -> "LatentsOutput":
         return cls(
             latents=LatentsField(latents_name=latents_name, seed=seed),
-            width=latents.size()[3] * 8,
-            height=latents.size()[2] * 8,
+            width=latents.size()[3] * LATENT_SCALE_FACTOR,
+            height=latents.size()[2] * LATENT_SCALE_FACTOR,
         )