From 067f88d9af3a8824f01a26c7bbbf1668d2cb6497 Mon Sep 17 00:00:00 2001 From: newsletternewsletter <168619624+newsletternewsletter@users.noreply.github.com> Date: Wed, 6 Aug 2025 16:44:18 +0200 Subject: [PATCH] refactor magic numbers with tile variable references Fixed the tile variable usage to replace magic numbers with tile variable references. --- comfy_extras/nodes_upscale_model.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/comfy_extras/nodes_upscale_model.py b/comfy_extras/nodes_upscale_model.py index 04c948341..c3d4591c9 100644 --- a/comfy_extras/nodes_upscale_model.py +++ b/comfy_extras/nodes_upscale_model.py @@ -49,18 +49,18 @@ class ImageUpscaleWithModel: def upscale(self, upscale_model, image): device = model_management.get_torch_device() - + + tile = 512 + overlap = 32 + memory_required = model_management.module_size(upscale_model.model) - memory_required += (512 * 512 * 3) * image.element_size() * max(upscale_model.scale, 1.0) * 384.0 #The 384.0 is an estimate of how much some of these models take, TODO: make it more accurate + memory_required += (tile * tile * 3) * image.element_size() * max(upscale_model.scale, 1.0) * 384.0 #The 384.0 is an estimate of how much some of these models take, TODO: make it more accurate memory_required += image.nelement() * image.element_size() model_management.free_memory(memory_required, device) upscale_model.to(device) in_img = image.movedim(-1,-3).to(device) - tile = 512 - overlap = 32 - oom = True while oom: try: