wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-22 20:30:25 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Fix kwargs

Browse Source
This commit is contained in:
ifilipis 2026-01-08 23:30:52 +00:00 committed by GitHub
parent c13ec6ac00
commit b93f165c2e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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/disk_weights.py
View File

@ -753,7 +753,7 @@ def ensure_module_materialized(
_log_materialization(module, target_device, free_mem_start, refs, state, "Disk weight materialized") _log_materialization(module, target_device, free_mem_start, refs, state, "Disk weight materialized")
def disk_weight_pre_hook(module: torch.nn.Module, args, kwargs): def disk_weight_pre_hook(module: torch.nn.Module, args, kwargs={}):
if not REGISTRY.has(module) and module not in LAZY_MODULE_STATE: if not REGISTRY.has(module) and module not in LAZY_MODULE_STATE:
return return
input_dtype = _find_tensor_dtype(args, kwargs) input_dtype = _find_tensor_dtype(args, kwargs)