Fix regression.

main
comfyanonymous 7 months ago
parent e172564eea
commit 6678d5cf65

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

Loading…
Cancel
Save