Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

SDNQ fix _tied_weights_keys is dict case

Browse Source
This commit is contained in:
Disty0
2025-11-25 19:37:46 +03:00
parent aeb71d172e
commit da3c439059
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sdnq/common.py
+1
View File
@@ -127,6 +127,7 @@ common_skip_keys = (
".img_out",
".vid_out",
".final_layer",
"multi_modal_projector",
"time_text_embed",
"patch_embedding",
"patch_embed",
modules/sdnq/quantizer.py
+5 -1
View File
@@ -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: