wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-02 17:50:19 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

nodes_model_patch: fix copy-paste coding error

Browse Source
This commit is contained in:
Rattus 2026-01-20 22:04:06 +10:00
parent b915d13e57
commit e36ffd2cee
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy_extras/nodes_model_patch.py
View File

@ -268,7 +268,7 @@ class ModelPatchLoader:
operations=comfy.ops.manual_cast) operations=comfy.ops.manual_cast)
model_patcher = comfy.model_patcher.CoreModelPatcher(model, load_device=comfy.model_management.get_torch_device(), offload_device=comfy.model_management.unet_offload_device()) model_patcher = comfy.model_patcher.CoreModelPatcher(model, load_device=comfy.model_management.get_torch_device(), offload_device=comfy.model_management.unet_offload_device())
model.load_state_dict(sd, assign=self.model_patcher.is_dynamic()) model.load_state_dict(sd, assign=model_patcher.is_dynamic())
return (model_patcher,) return (model_patcher,)