From f45e3342e6fb6f05d5a90033135e4140cb741cfc Mon Sep 17 00:00:00 2001 From: Disty0 Date: Mon, 11 Aug 2025 15:11:29 +0300 Subject: [PATCH] Cleanup --- modules/sdnq/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modules/sdnq/__init__.py b/modules/sdnq/__init__.py index b6e6deb52..e1c8396f9 100644 --- a/modules/sdnq/__init__.py +++ b/modules/sdnq/__init__.py @@ -412,6 +412,8 @@ class SDNQConfig(QuantizationConfigMixin): Same as use_quantized_matmul_conv but for the convolutional layers with UNets like SDXL. dequantize_fp32 (`bool`, *optional*, defaults to `False`): Enabling this option will use FP32 on the dequantization step. + non_blocking (`bool`, *optional*, defaults to `False`): + Enabling this option will use non blocking ops when moving layers between the quantization device and the return device. quantization_device (`torch.device`, *optional*, defaults to `None`): Used to set which device will be used for the quantization calculation on model load. return_device (`torch.device`, *optional*, defaults to `None`):