Adding SD3

model/model_registry.py

@@ -94,6 +94,13 @@ register_model_info(
     "SDXL is a Latent Diffusion Model that uses two fixed, pretrained text encoders.",
 )
 
+register_model_info(
+    ["imagenhub_SD3_generation"],
+    "SD3",
+    "https://huggingface.co/blog/sd3",
+    "SD3 is a novel Multimodal Diffusion Transformer (MMDiT) model.",
+)
+
 register_model_info(
     ["imagenhub_PixArtAlpha_generation"],
     "PixArtAlpha",

model/models/__init__.py

@@ -9,7 +9,7 @@ from .videogenhub_models import load_videogenhub_model
 #                             'playground_PlayGroundV2_generation', 'playground_PlayGroundV2.5_generation']
 IMAGE_GENERATION_MODELS = ['imagenhub_SDXLTurbo_generation','imagenhub_SDXL_generation', 'imagenhub_PixArtAlpha_generation', 'imagenhub_PixArtSigma_generation',
                            'imagenhub_OpenJourney_generation','imagenhub_SDXLLightning_generation', 'imagenhub_StableCascade_generation', 'imagenhub_HunyuanDiT_generation',
-                           'playground_PlayGroundV2.5_generation']  # 'playground_PlayGroundV2_generation'
+                           'playground_PlayGroundV2.5_generation', 'imagenhub_SD3_generation']  # 'playground_PlayGroundV2_generation'
 IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
                         'imagenhub_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition', 
                         'imagenhub_MagicBrush_edition', 'imagenhub_PNP_edition',