From 975361b40491b2bb875ae6d67774416cbe83a901 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Sat, 25 Oct 2025 20:06:13 +0300 Subject: [PATCH] import sdnq before load --- pipelines/generic.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/pipelines/generic.py b/pipelines/generic.py index 0edd9d2cd..784ea3b1b 100644 --- a/pipelines/generic.py +++ b/pipelines/generic.py @@ -12,6 +12,8 @@ def load_transformer(repo_id, cls_name, load_config={}, subfolder="transformer", transformer = None jobid = shared.state.begin('Load DiT') try: + if 'sdnq-' not in repo_id.lower(): + from modules import sdnq # register to diffusers and transformers load_args, quant_args = model_quant.get_dit_args(load_config, module='Model', device_map=True, allow_quant=allow_quant, modules_to_not_convert=modules_to_not_convert, modules_dtype_dict=modules_dtype_dict) quant_type = model_quant.get_quant_type(quant_args) dtype = dtype or devices.dtype @@ -83,6 +85,8 @@ def load_text_encoder(repo_id, cls_name, load_config={}, subfolder="text_encoder text_encoder = None jobid = shared.state.begin('Load TE') try: + if 'sdnq-' not in repo_id.lower(): + from modules import sdnq # register to diffusers and transformers load_args, quant_args = model_quant.get_dit_args(load_config, module='TE', device_map=True, allow_quant=allow_quant, modules_to_not_convert=modules_to_not_convert, modules_dtype_dict=modules_dtype_dict) quant_type = model_quant.get_quant_type(quant_args) dtype = dtype or devices.dtype