fix model type

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-04-24 13:18:45 -04:00
parent b15a93b729
commit 2b4962182f
+1 -1
View File
@@ -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'