From 2b4962182ff6d7c7f219c0e620d8af6b8f4de9db Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Thu, 24 Apr 2025 13:18:45 -0400
Subject: [PATCH] fix model type

Signed-off-by: Vladimir Mandic <mandic00@live.com>
---
 modules/modeldata.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/modeldata.py b/modules/modeldata.py
index ae3e560ad..bc9e2519f 100644
--- a/modules/modeldata.py
+++ b/modules/modeldata.py
@@ -27,7 +27,7 @@ def get_model_type(pipe):
         model_type = 'sc'
     elif "AuraFlow" in name:
         model_type = 'auraflow'
-    elif "Flux" in name or "Flex.1" or "Flex.2":
+    elif "Flux" in name or "Flex.1" in name or "Flex.2" in name:
         model_type = 'f1'
     elif "Lumina2" in name:
         model_type = 'lumina2'