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

Better fix for lowvram issue.

Browse Source
This commit is contained in:
comfyanonymous 2024-08-09 12:16:25 -04:00
parent 86a97e91fc
commit a3cc326748
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_patcher.py
View File

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