autodetect fallback by name

2026-09-19 17:24:32 +02:00 · 2024-01-24 11:37:55 -05:00
parent 9fcc196583
commit dad7fe7dcf
1 changed files with 4 additions and 1 deletions
@@ -577,7 +577,10 @@ def detect_pipeline(f: str, op: str = 'model', warning=True):
                    warn(f'Model detected as Segmind Vega model, but attempting to load using backend=original: {op}={f} size={size} MB')
                guess = 'Stable Diffusion XL'
            else:
-                guess = 'Stable Diffusion'
+                if 'XL' in f.upper():
+                    guess = 'Stable Diffusion XL'
+                else:
+                    guess = 'Stable Diffusion'
            # guess by name
            """
            if 'LCM_' in f.upper() or 'LCM-' in f.upper() or '_LCM' in f.upper() or '-LCM' in f.upper():