From 064b64c76c790b8cdccf97e1503e08eabce50c39 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Mon, 8 Dec 2025 01:14:19 +0300 Subject: [PATCH] cleanup --- modules/sdnq/loader.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/modules/sdnq/loader.py b/modules/sdnq/loader.py index 818bff90d..205a99072 100644 --- a/modules/sdnq/loader.py +++ b/modules/sdnq/loader.py @@ -147,12 +147,14 @@ def post_process_model(model): for module_name, module in model.named_children(): if hasattr(module, "sdnq_dequantizer"): module.weight.requires_grad_(False) - if module.sdnq_dequantizer.use_quantized_matmul and not module.sdnq_dequantizer.re_quantize_for_matmul: - module.weight.data = prepare_weight_for_matmul(module.weight) + module.scale.requires_grad_(False) if module.zero_point is not None: module.zero_point.requires_grad_(False) + if module.sdnq_dequantizer.use_quantized_matmul and not module.sdnq_dequantizer.re_quantize_for_matmul: + module.weight.data = prepare_weight_for_matmul(module.weight) if module.svd_up is not None: - module.svd_up, module.svd_down = module.svd_up.requires_grad_(False), module.svd_down.requires_grad_(False) + module.svd_up.requires_grad_(False) + module.svd_down.requires_grad_(False) module.svd_up.data, module.svd_down.data = prepare_svd_for_matmul(module.svd_up, module.svd_down, module.sdnq_dequantizer.use_quantized_matmul) setattr(model, module_name, module) else: