From da3c4390597cc8fa151e14753104c5344af9cc85 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Tue, 25 Nov 2025 19:37:46 +0300 Subject: [PATCH] SDNQ fix _tied_weights_keys is dict case --- modules/sdnq/common.py | 1 + modules/sdnq/quantizer.py | 6 +++++- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/modules/sdnq/common.py b/modules/sdnq/common.py index c28bbf831..cefa87a9c 100644 --- a/modules/sdnq/common.py +++ b/modules/sdnq/common.py @@ -127,6 +127,7 @@ common_skip_keys = ( ".img_out", ".vid_out", ".final_layer", + "multi_modal_projector", "time_text_embed", "patch_embedding", "patch_embed", diff --git a/modules/sdnq/quantizer.py b/modules/sdnq/quantizer.py index 801642a0a..eddce4fad 100644 --- a/modules/sdnq/quantizer.py +++ b/modules/sdnq/quantizer.py @@ -165,7 +165,11 @@ def add_module_skip_keys(model, modules_to_not_convert: List[str] = None, module if getattr(model, "_keep_in_fp32_modules", None) is not None: modules_to_not_convert.extend(model._keep_in_fp32_modules) # pylint: disable=protected-access if getattr(model, "_tied_weights_keys", None) is not None: - modules_to_not_convert.extend(model._tied_weights_keys) # pylint: disable=protected-access + if isinstance(model._tied_weights_keys, dict): # pylint: disable=protected-access + modules_to_not_convert.extend(model._tied_weights_keys.keys()) # pylint: disable=protected-access + modules_to_not_convert.extend(model._tied_weights_keys.values()) # pylint: disable=protected-access + else: + modules_to_not_convert.extend(model._tied_weights_keys) # pylint: disable=protected-access skip_key_list = module_skip_keys_dict.get(model.__class__.__name__, None) if skip_key_list is not None: