From 3ae7ecdbad80f18ff6ae70a14317c0bdc2bd93d3 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Mon, 8 Dec 2025 01:29:52 +0300 Subject: [PATCH] SDNQ fix quantization_device getting ignored on post load quant --- modules/sdnq/quantizer.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/modules/sdnq/quantizer.py b/modules/sdnq/quantizer.py index 73b3895a1..bad764271 100644 --- a/modules/sdnq/quantizer.py +++ b/modules/sdnq/quantizer.py @@ -396,11 +396,10 @@ def sdnq_quantize_layer(layer, weights_dtype="int8", quantized_matmul_dtype=None return layer use_quantized_matmul = use_quantized_matmul_conv - layer.weight.requires_grad = False + layer.weight.requires_grad_(False) if return_device is None: return_device = layer.weight.device - if quantization_device is None: - quantization_device = layer.weight.device + if quantization_device is not None: layer.weight.data = layer.weight.to(quantization_device, non_blocking=non_blocking) (