jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

Fix issue with some lowvram weights.

Browse Source
This commit is contained in:
comfyanonymous
2024-08-09 03:58:28 -04:00
parent 55ad9d5f8c
commit 5acdadc9f3
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/model_patcher.py
View File

@@ -393,10 +393,13 @@ class ModelPatcher:
if m.comfy_cast_weights:
wipe_lowvram_weight(m)
if hasattr(m, "weight"):
param = list(m.parameters())
if len(param) > 0:
mem_counter += comfy.model_management.module_size(m)
if m.weight is not None and m.weight.device == device_to:
weight = param[0]
if weight.device == device_to:
continue
self.patch_weight_to_device(weight_key) #TODO: speed this up without OOM
self.patch_weight_to_device(bias_key)
m.to(device_to)