wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-10 13:32:36 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

Merge branch 'comfyanonymous:master' into refactor/onprompt

Browse Source
This commit is contained in:
Dr.Lt.Data 2023-07-04 15:31:17 +09:00 committed by GitHub
commit 5de7d30bea
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
comfy/model_management.py
View File

@ -456,7 +456,13 @@ def mps_mode():
def is_device_cpu(device):
if hasattr(device, 'type'):
if (device.type == 'cpu' or device.type == 'mps'):
if (device.type == 'cpu'):
return True
return False
def is_device_mps(device):
if hasattr(device, 'type'):
if (device.type == 'mps'):
return True
return False
@ -468,7 +474,7 @@ def should_use_fp16(device=None, model_params=0):
return True
if device is not None: #TODO
if is_device_cpu(device):
if is_device_cpu(device) or is_device_mps(device):
return False
if FORCE_FP32:

4
comfy_extras/nodes_model_merging.py
View File

@ -46,9 +46,11 @@ class ModelMergeBlocks:
ratio = default_ratio
k_unet = k[len("diffusion_model."):]
last_arg_size = 0
for arg in kwargs:
if k_unet.startswith(arg):
if k_unet.startswith(arg) and last_arg_size < len(arg):
ratio = kwargs[arg]
last_arg_size = len(arg)
m.add_patches({k: (sd[k], )}, 1.0 - ratio, ratio)
return (m, )