From e06cbea7aa33e8be0f45aba23db1c0160b299243 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Wed, 28 May 2025 15:55:08 +0300 Subject: [PATCH] Cleanup --- modules/model_quant_sdnq.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/modules/model_quant_sdnq.py b/modules/model_quant_sdnq.py index 771029069..022b132e4 100644 --- a/modules/model_quant_sdnq.py +++ b/modules/model_quant_sdnq.py @@ -96,10 +96,10 @@ def sdnq_quantize_layer(layer, weights_dtype="int8", torch_dtype=None, group_siz if shared.opts.diffusers_offload_mode in {"none", "model"}: return_device = devices.device elif pre_mode: - if shared.opts.sdnq_quantize_with_gpu: - return_device = devices.cpu - elif shared.opts.device_map == "gpu": + if shared.opts.device_map == "gpu": return_device = devices.device + elif shared.opts.sdnq_quantize_with_gpu: + return_device = devices.cpu else: return_device = layer.weight.device else: