wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 23:00:51 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Be less strict when loading mixed ops weights. (#11769)

Browse Source
This commit is contained in:
comfyanonymous 2026-01-09 11:21:06 -08:00 committed by GitHub
parent ec0a832acb
commit bd0e6825e8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/ops.py
View File

@ -546,7 +546,8 @@ def mixed_precision_ops(quant_config={}, compute_dtype=torch.bfloat16, full_prec
weight_key = f"{prefix}weight" weight_key = f"{prefix}weight"
weight = state_dict.pop(weight_key, None) weight = state_dict.pop(weight_key, None)
if weight is None: if weight is None:
raise ValueError(f"Missing weight for layer {layer_name}") logging.warning(f"Missing weight for layer {layer_name}")
return
manually_loaded_keys = [weight_key] manually_loaded_keys = [weight_key]