From b93f165c2e9d14610d07cb8af3149a2f1ca6b79f Mon Sep 17 00:00:00 2001 From: ifilipis <40601736+ifilipis@users.noreply.github.com> Date: Thu, 8 Jan 2026 23:30:52 +0000 Subject: [PATCH] Fix kwargs --- comfy/disk_weights.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/disk_weights.py b/comfy/disk_weights.py index fb9186db4..07e9b5f11 100644 --- a/comfy/disk_weights.py +++ b/comfy/disk_weights.py @@ -753,7 +753,7 @@ def ensure_module_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: return input_dtype = _find_tensor_dtype(args, kwargs)