Skip to content

Commit 9eaba63

Browse files
Fix upscale models breaking on non dynamic vram low vram. (Comfy-Org#15437)
1 parent 00d02f2 commit 9eaba63

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

comfy_extras/nodes_upscale_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ def execute(cls, upscale_model, image) -> io.NodeOutput:
7272

7373
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
7474
memory_required += image.nelement() * image.element_size()
75-
model_management.load_models_gpu([upscale_model.patcher], memory_required=memory_required)
75+
model_management.load_models_gpu([upscale_model.patcher], memory_required=memory_required, force_full_load=True)
7676

7777
in_img = image.movedim(-1,-3).to(device)
7878

0 commit comments

Comments
 (0)