oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-01-04 15:50:26 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

More precise pipeline type discovery

Browse source
This commit is contained in:
oobabooga 2025-12-02 04:50:33 -08:00
parent 07754f9f29
commit 83af6864d5
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
modules/image_models.py
View file

@ -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'