cleanup model types

Signed-off-by: vladmandic <mandic00@live.com>
This commit is contained in:
vladmandic
2026-01-15 08:30:48 +01:00
parent 387b3c7c36
commit 71d3482168
6 changed files with 25 additions and 24 deletions
+1 -1
View File
@@ -439,7 +439,7 @@ def load_diffuser_force(detected_model_type, checkpoint_info, diffusers_load_con
from pipelines.model_kandinsky import load_kandinsky3
sd_model = load_kandinsky3(checkpoint_info, diffusers_load_config)
allow_post_quant = False
elif model_type in ['Kandinsky 5.0']:
elif model_type in ['Kandinsky 5.0'] and '2I' in model_type:
from pipelines.model_kandinsky import load_kandinsky5
sd_model = load_kandinsky5(checkpoint_info, diffusers_load_config)
allow_post_quant = False