add lumina2

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-02-12 08:54:00 -05:00
parent 1d533544d2
commit a7ccea60ff
13 changed files with 78 additions and 11 deletions
+4
View File
@@ -69,6 +69,8 @@ def detect_pipeline(f: str, op: str = 'model', warning=True, quiet=False):
guess = 'Sana'
if 'lumina-next' in f.lower():
guess = 'Lumina-Next'
if 'lumina-image-2' in f.lower():
guess = 'Lumina2'
if 'kolors' in f.lower():
guess = 'Kolors'
if 'auraflow' in f.lower():
@@ -101,6 +103,8 @@ def detect_pipeline(f: str, op: str = 'model', warning=True, quiet=False):
guess = 'FLUX'
if 'StableDiffusion3' in pipeline.__name__:
guess = 'Stable Diffusion 3'
if 'Lumina2' in pipeline.__name__:
guess = 'Lumina 2'
# switch for specific variant
if guess == 'Stable Diffusion' and 'inpaint' in f.lower():
guess = 'Stable Diffusion Inpaint'