From 76b00176757417c0b892429f6dd5f57bca8d6eee Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Tue, 5 Mar 2024 21:44:20 -0500 Subject: [PATCH] fix move model --- extensions-builtin/sd-webui-controlnet | 2 +- modules/sd_models.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/extensions-builtin/sd-webui-controlnet b/extensions-builtin/sd-webui-controlnet index 489a399ef..4b5a42f26 160000 --- a/extensions-builtin/sd-webui-controlnet +++ b/extensions-builtin/sd-webui-controlnet @@ -1 +1 @@ -Subproject commit 489a399ef06cb94cc8c580ecbf91c5fa1bcce111 +Subproject commit 4b5a42f265594c00b26937b947630895d632a9f7 diff --git a/modules/sd_models.py b/modules/sd_models.py index 9074f6bda..1ac2354a9 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -758,7 +758,7 @@ def move_model(model, device=None, force=False): if hasattr(model, "components"): # accelerate patch for name, m in model.components.items(): if not hasattr(m, "_hf_hook"): # not accelerate hook - return + break if not isinstance(m, torch.nn.Module) or name in model._exclude_from_cpu_offload: # pylint: disable=protected-access continue for module in m.modules():