From 83af6864d59de7cdd1f41809f53d02c0a008a66e Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Tue, 2 Dec 2025 04:50:33 -0800
Subject: [PATCH] More precise pipeline type discovery

---
 modules/image_models.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/modules/image_models.py b/modules/image_models.py
index e4831758..e6f9a172 100644
--- a/modules/image_models.py
+++ b/modules/image_models.py
@@ -83,9 +83,9 @@ def get_pipeline_type(pipe):
         str: 'zimage', 'qwenimage', or 'unknown'
     """
     class_name = pipe.__class__.__name__
-    if 'ZImage' in class_name:
+    if class_name == 'ZImagePipeline':
         return 'zimage'
-    elif 'QwenImage' in class_name:
+    elif class_name == 'QwenImagePipeline':
         return 'qwenimage'
     else:
         return 'unknown'