From 1cc502313e46769efbd550ff391ae3efaa190dba Mon Sep 17 00:00:00 2001
From: Thomas Ward <teward@thomas-ward.net>
Date: Fri, 24 May 2024 02:01:59 -0400
Subject: [PATCH] Use proper parameters for width, height.

---
 generation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/generation.py b/generation.py
index ed88a57..580e11d 100644
--- a/generation.py
+++ b/generation.py
@@ -147,7 +147,7 @@ class TWCUI_Util_GenerationParameters(BaseNode):
             seed = new_random_seed()
 
         batch_size = 1
-        LATENT = torch.zeros([batch_size, 4, height // 8, width // 8], device=self.device)
+        LATENT = torch.zeros([batch_size, 4, image_height // 8, image_width // 8], device=self.device)
 
         return (MODEL, CLIP, VAE, LATENT, image_width, image_height, sampling_steps, cfg, sampler_name, scheduler_name,
                 seed)