add lumina-dimoo

Signed-off-by: vladmandic <mandic00@live.com>

modules/caption/models_def.py

@@ -7,6 +7,7 @@ vlm_models = {
     "Google Gemma 3n E4B": "google/gemma-3n-E4B-it", # 1.5GB
     "Google Gemma 4 E2B": "google/gemma-4-E2B-it",
     "Google Gemma 4 E4B": "google/gemma-4-E4B-it",
+    "Heretic Gemma 4 E4B": "p-e-w/gemma-4-E2B-it-heretic-ara",
     "Nidum Gemma 3 4B Uncensored": "nidum/Nidum-Gemma-3-4B-it-Uncensored",
     "Allura Gemma 3 Glitter 4B": "allura-org/Gemma-3-Glitter-4B",
     # Qwen3.5

modules/modeldata.py

@@ -52,6 +52,8 @@ def get_model_type(pipe):
         model_type = 'f1'
     elif "ZImage" in name or "Z-Image" in name:
         model_type = 'zimage'
+    elif "LuminaDiMOO" in name:
+        model_type = 'luminadimoo'
     elif "Lumina2" in name:
         model_type = 'lumina2'
     elif "Lumina" in name:

modules/sd_detect.py

@@ -67,6 +67,8 @@ def guess_by_name(fn, current_guess):
         new_guess = 'Sana'
     elif 'lumina-next' in fn.lower():
         new_guess = 'Lumina-Next'
+    elif 'lumina-dimoo' in fn.lower():
+        new_guess = 'Lumina-DiMOO'
     elif 'lumina-image-2' in fn.lower():
         new_guess = 'Lumina 2'
     elif 'kolors' in fn.lower():

modules/sd_models.py

@@ -361,6 +361,10 @@ def load_diffuser_force(detected_model_type, checkpoint_info, diffusers_load_con
             from pipelines.model_lumina import load_lumina
             sd_model = load_lumina(checkpoint_info, diffusers_load_config)
             allow_post_quant = True
+        elif model_type in ['Lumina-DiMOO']:
+            from pipelines.model_lumina import load_lumina_dimoo
+            sd_model = load_lumina_dimoo(checkpoint_info, diffusers_load_config)
+            allow_post_quant = False
         elif model_type in ['Kolors']:
             from pipelines.model_kolors import load_kolors
             sd_model = load_kolors(checkpoint_info, diffusers_load_config)

modules/shared_items.py

@@ -30,6 +30,7 @@ pipelines = {
     'Chroma': getattr(diffusers, 'ChromaPipeline', None),
     'Sana': getattr(diffusers, 'SanaPipeline', None),
     'Lumina-Next': getattr(diffusers, 'LuminaText2ImgPipeline', None),
+    'Lumina-DiMOO': getattr(diffusers, 'LuminaDiMOOPipeline', None),
     'Lumina 2': getattr(diffusers, 'Lumina2Pipeline', None),
     'AuraFlow': getattr(diffusers, 'AuraFlowPipeline', None),
     'Kandinsky 2.1': getattr(diffusers, 'KandinskyCombinedPipeline', None),