Skip to content

Commit 89a84e3

Browse files
Disable initial GPU load when novram is used. (Comfy-Org#8294)
1 parent e5799c4 commit 89a84e3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy/model_management.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -695,7 +695,7 @@ def unet_inital_load_device(parameters, dtype):
695695
return torch_dev
696696

697697
cpu_dev = torch.device("cpu")
698-
if DISABLE_SMART_MEMORY:
698+
if DISABLE_SMART_MEMORY or vram_state == VRAMState.NO_VRAM:
699699
return cpu_dev
700700

701701
model_size = dtype_size(dtype) * parameters

0 commit comments

Comments
 (0)