llada t2i

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

modules/modeldata.py

@@ -119,6 +119,8 @@ def get_model_type(pipe):
         model_type = 'longcat'
     elif 'GlmImage' in name:
         model_type = 'glmimage'
+    elif 'LLaDAImage' in name:
+        model_type = 'lladaimage'
     elif 'Step1XEdit' in name:
         model_type = 'step1x_edit'
     elif 'JoyImageEdit' in name:

modules/sd_detect.py

@@ -167,6 +167,8 @@ def guess_by_name(fn, current_guess):
         new_guess = 'OvisImage'
     elif 'glm-image' in fn.lower():
         new_guess = 'GLMImage'
+    elif 'llada' in fn.lower():
+        new_guess = 'LLaDAImage'
     elif 'sdxs-1b' in fn.lower():
         new_guess = 'SDXS'
     elif 'step1x-edit' in fn.lower():

modules/sd_models.py

@@ -55,6 +55,7 @@ pipe_switch_task_exclude = [
     'Kandinsky5I2IPipeline',
     'GoogleNanoBananaPipeline',
     'Step1XEditPipeline',
+    'LLaDAImagePipeline',
     'BooguImagePipeline',
     'BooguImageTurboPipeline',
 ]
@@ -610,6 +611,10 @@ def load_diffuser_force(detected_model_type: str, checkpoint_info: CheckpointInf
             from pipelines.model_glm import load_glm_image
             sd_model = load_glm_image(checkpoint_info, diffusers_load_config)
             allow_post_quant = False
+        elif model_type in ['LLaDAImage']:
+            from pipelines.model_llada import load_llada_image
+            sd_model = load_llada_image(checkpoint_info, diffusers_load_config)
+            allow_post_quant = False
         elif model_type in ['SDXS']:
             from pipelines.model_sdxs import load_sdxs
             sd_model = load_sdxs(checkpoint_info, diffusers_load_config)

modules/vae/sd_vae_taesd.py

@@ -73,7 +73,7 @@ def get_model(model_cls, variant=None):
     elif model_cls in {'f1', 'h1', 'zimage', 'lumina2', 'chroma', 'longcat', 'omnigen2', 'flite', 'ovis', 'kandinsky5', 'glmimage', 'cogview3', 'cogview4', 'ultraflux'}:
         model_cls = 'f1'
         variant = 'TAE FLUX.1'
-    elif model_cls in {'f2', 'ernieimage', 'lens', 'ideogram4'}:
+    elif model_cls in {'f2', 'ernieimage', 'lens', 'ideogram4', 'lladaimage'}:
         model_cls = 'f2'
         variant = 'TAE FLUX.2'
     elif model_cls in {'sd3'}: