Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix move model

Browse Source
This commit is contained in:
Vladimir Mandic
2024-03-05 21:44:20 -05:00
parent 9054b5cf79
commit 76b0017675
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extensions-builtin/sd-webui-controlnet
+1 -1
Submodule extensions-builtin/sd-webui-controlnet updated: 489a399ef0...4b5a42f265
modules/sd_models.py
+1 -1
View File
@@ -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():