wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-10 14:20:49 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Merge 14700174a5 into 5943fbf457

Browse Source
This commit is contained in:
newsletternewsletter 2026-01-08 16:37:32 +00:00 committed by GitHub
commit bb2203d095
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: def execute(cls, upscale_model, image) -> io.NodeOutput:
device = model_management.get_torch_device() device = model_management.get_torch_device()
tile = 512
overlap = 32
memory_required = model_management.module_size(upscale_model.model) 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() memory_required += image.nelement() * image.element_size()
model_management.free_memory(memory_required, device) model_management.free_memory(memory_required, device)
upscale_model.to(device) upscale_model.to(device)
in_img = image.movedim(-1,-3).to(device) in_img = image.movedim(-1,-3).to(device)
tile = 512
overlap = 32
oom = True oom = True
try: try:
while oom: while oom: