xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-06-06 07:33:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix regression.

Browse Source
This commit is contained in:
comfyanonymous 2024-08-09 14:02:38 -04:00
parent e172564eea
commit 6678d5cf65
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/model_patcher.py
View File

@ -396,6 +396,7 @@ class ModelPatcher:
if hasattr(m, "weight"): if hasattr(m, "weight"):
mem_counter += comfy.model_management.module_size(m) mem_counter += comfy.model_management.module_size(m)
param = list(m.parameters()) param = list(m.parameters())
if len(param) > 0:
weight = param[0] weight = param[0]
if weight.device == device_to: if weight.device == device_to:
continue continue