From 9c16e2234a83dd68b0b97fccb9afe345d59b878f Mon Sep 17 00:00:00 2001 From: Disty0 Date: Wed, 8 Oct 2025 01:18:12 +0300 Subject: [PATCH] cleanup --- modules/sdnq/quantizer.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/modules/sdnq/quantizer.py b/modules/sdnq/quantizer.py index 190ff096f..50d4b49f3 100644 --- a/modules/sdnq/quantizer.py +++ b/modules/sdnq/quantizer.py @@ -322,6 +322,10 @@ def apply_sdnq_to_module(model, weights_dtype="int8", torch_dtype=None, group_si def add_module_skip_keys(model, modules_to_not_convert: List[str] = None, modules_dtype_dict: Dict[str, List[str]] = None): + if modules_to_not_convert is None: + modules_to_not_convert = [] + if modules_dtype_dict is None: + modules_dtype_dict = {} 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, "_skip_layerwise_casting_patterns", None) is not None: @@ -356,11 +360,6 @@ def sdnq_post_load_quant( modules_dtype_dict: Dict[str, List[str]] = None, op=None, ): - if modules_to_not_convert is None: - modules_to_not_convert = [] - if modules_dtype_dict is None: - modules_dtype_dict = {} - if add_skip_keys: model, modules_to_not_convert, modules_dtype_dict = add_module_skip_keys(model, modules_to_not_convert, modules_dtype_dict)