wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-30 21:43:43 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

Merge 19eb196941 into a500f1edac

Browse Source
This commit is contained in:
isaac-bender 2026-03-29 15:00:26 +00:00 committed by GitHub
commit 045bdb2c9c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
comfy/model_patcher.py
View File

@ -689,7 +689,15 @@ class ModelPatcher:
inplace_update = self.weight_inplace_update or inplace_update
if key not in self.backup and not return_weight:
self.backup[key] = collections.namedtuple('Dimension', ['weight', 'inplace_update'])(weight.to(device=self.offload_device, copy=inplace_update), inplace_update)
# When set_func is present (e.g. QuantizedTensor/fp8_scaled ops), it replaces
# the parameter object rather than modifying it in-place. The restore path
# must therefore also replace the parameter (set_attr_param) instead of doing
# an in-place copy (copy_to_param), because QuantizedTensor.__torch_dispatch__
# routes copy_() through dequant-and-fallback which silently fails to update
# the underlying quantized data. Force inplace_update=False in the backup
# for these keys so unpatch_model uses set_attr_param for restoration.
backup_inplace = inplace_update if set_func is None else False
self.backup[key] = collections.namedtuple('Dimension', ['weight', 'inplace_update'])(weight.to(device=self.offload_device, copy=inplace_update), backup_inplace)
temp_dtype = comfy.model_management.lora_compute_dtype(device_to)
if device_to is not None: