From 331260d90840187364e4bf17671bdc361085d606 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 6 Aug 2025 09:45:54 +0300
Subject: [PATCH] Update model_optimization_nodes.py

---
 nodes/model_optimization_nodes.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/nodes/model_optimization_nodes.py b/nodes/model_optimization_nodes.py
index c3fd719..dcf828d 100644
--- a/nodes/model_optimization_nodes.py
+++ b/nodes/model_optimization_nodes.py
@@ -19,7 +19,10 @@ if not _initialized:
     _original_functions["orig_attention"] = comfy_attention.optimized_attention
     _original_functions["original_patch_model"] = comfy.model_patcher.ModelPatcher.patch_model
     _original_functions["original_load_lora_for_models"] = comfy.sd.load_lora_for_models
-    _original_functions["original_qwen_forward"] = comfy.ldm.qwen_image.model.Attention.forward
+    try:
+        _original_functions["original_qwen_forward"] = comfy.ldm.qwen_image.model.Attention.forward
+    except:
+        pass
     _initialized = True
 
 class BaseLoaderKJ: