From 7eaa203aabd46ecb95d3b81ff09f06ed8f8b499a Mon Sep 17 00:00:00 2001 From: ussoewwin <136552381+ussoewwin@users.noreply.github.com> Date: Sun, 25 Jan 2026 18:11:57 +0900 Subject: [PATCH] V3: Always enable AWQ modulation layer LoRA (no switch needed) --- nodes/lora/qwenimage_v3.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/nodes/lora/qwenimage_v3.py b/nodes/lora/qwenimage_v3.py index 593e5dd..aebdb80 100644 --- a/nodes/lora/qwenimage_v3.py +++ b/nodes/lora/qwenimage_v3.py @@ -207,12 +207,16 @@ class NunchakuQwenImageLoraStackV2: if model_wrapper.cpu_offload_setting != cpu_offload: logger.info(f"🔄 Updating CPU offload setting from '{model_wrapper.cpu_offload_setting}' to '{cpu_offload}'") model_wrapper.cpu_offload_setting = cpu_offload + # V3: Always ensure apply_awq_mod is True (no switch, always enabled) + if hasattr(model_wrapper, "apply_awq_mod") and model_wrapper.apply_awq_mod != True: + logger.info(f"🔄 Updating AWQ mod setting from '{model_wrapper.apply_awq_mod}' to 'True' (V3: always enabled)") + model_wrapper.apply_awq_mod = True transformer = model_wrapper.model elif model_wrapper_type_name == "NunchakuQwenImageTransformer2DModel" or model_wrapper_type_name.endswith( "NunchakuQwenImageTransformer2DModel" ): logger.info("🔧 Wrapping NunchakuQwenImageTransformer2DModel with ComfyQwenImageWrapper") - logger.info(f"📦 Creating ComfyQwenImageWrapper with cpu_offload='{cpu_offload}'") + logger.info(f"📦 Creating ComfyQwenImageWrapper with cpu_offload='{cpu_offload}', apply_awq_mod=True (V3: always enabled)") wrapped_model = ComfyQwenImageWrapper( model_wrapper, getattr(model_wrapper, "config", {}), @@ -220,6 +224,7 @@ class NunchakuQwenImageLoraStackV2: {}, # forward_kwargs cpu_offload, # cpu_offload_setting 4.0, # vram_margin_gb + apply_awq_mod=True, # V3: Always enable AWQ modulation layer LoRA (no switch needed) ) model.model.diffusion_model = wrapped_model model_wrapper = wrapped_model