298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 02:15:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Set initial load devices for te and model to mps device on mac.

Browse Source
This commit is contained in:
comfyanonymous 2024-12-12 06:00:31 -05:00
parent 3dfdddcc91
commit fd5dfb812c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/model_management.py
View File

@ -581,7 +581,7 @@ def unet_offload_device():
def unet_inital_load_device(parameters, dtype): def unet_inital_load_device(parameters, dtype):
torch_dev = get_torch_device() torch_dev = get_torch_device()
if vram_state == VRAMState.HIGH_VRAM: if vram_state == VRAMState.HIGH_VRAM or vram_state == VRAMState.SHARED:
return torch_dev return torch_dev
cpu_dev = torch.device("cpu") cpu_dev = torch.device("cpu")
@ -695,7 +695,7 @@ def text_encoder_initial_device(load_device, offload_device, model_size=0):
return offload_device return offload_device
if is_device_mps(load_device): if is_device_mps(load_device):
return offload_device return load_device
mem_l = get_free_memory(load_device) mem_l = get_free_memory(load_device)
mem_o = get_free_memory(offload_device) mem_o = get_free_memory(offload_device)