Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-10 06:48:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

SDNQ rename unused param_name arg to op

Browse Source
This commit is contained in:
Disty0
2025-08-13 22:10:30 +03:00
parent 4d95cf47ce
commit 8ca74d0cd2
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/model_quant.py
+1 -1
View File
@@ -428,9 +428,9 @@ def sdnq_quantize_model(model, op=None, sd_model=None, do_gc: bool = True, weigh
non_blocking=shared.opts.diffusers_offload_nonblocking,
quantization_device=quantization_device,
return_device=return_device,
param_name=op,
modules_to_not_convert=modules_to_not_convert,
modules_dtype_dict=modules_dtype_dict,
op=op,
)
t1 = time.time()
timer.load.add('sdnq', t1 - t0)
modules/sdnq/__init__.py
+2 -2
View File
@@ -186,7 +186,7 @@ def sdnq_quantize_layer(layer, weights_dtype="int8", torch_dtype=None, group_siz
return layer
def apply_sdnq_to_module(model, weights_dtype="int8", torch_dtype=None, group_size=0, quant_conv=False, use_quantized_matmul=False, use_quantized_matmul_conv=False, dequantize_fp32=False, non_blocking=False, quantization_device=None, return_device=None, param_name=None, modules_to_not_convert: List[str] = [], modules_dtype_dict: Dict[str, List[str]] = {}): # pylint: disable=unused-argument
def apply_sdnq_to_module(model, weights_dtype="int8", torch_dtype=None, group_size=0, quant_conv=False, use_quantized_matmul=False, use_quantized_matmul_conv=False, dequantize_fp32=False, non_blocking=False, quantization_device=None, return_device=None, modules_to_not_convert: List[str] = [], modules_dtype_dict: Dict[str, List[str]] = {}, op=None): # pylint: disable=unused-argument
has_children = list(model.children())
if not has_children:
return model
@@ -238,9 +238,9 @@ def apply_sdnq_to_module(model, weights_dtype="int8", torch_dtype=None, group_si
non_blocking=non_blocking,
quantization_device=quantization_device,
return_device=return_device,
param_name=param_name,
modules_to_not_convert=modules_to_not_convert,
modules_dtype_dict=modules_dtype_dict,
op=op,
)
return model