xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-07-13 04:16:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge 14700174a5ef2f1de7efead4a69b198126b251d8 into 9304e47351be8d178a093b30bcaf5d72c3a2baf5

Browse Source
This commit is contained in:
newsletternewsletter 2025-12-16 17:04:07 +08:00 committed by GitHub
commit 222c866caa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_extras/nodes_upscale_model.py
View File

@ -66,17 +66,17 @@ class ImageUpscaleWithModel(io.ComfyNode):
def execute(cls, upscale_model, image) -> io.NodeOutput:
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: