adding Kolors

model/model_registry.py

@@ -136,6 +136,13 @@ register_model_info(
     "HunyuanDiT is a Powerful Multi-Resolution Diffusion Transformer with Fine-Grained Chinese Understanding",
 )
 
+register_model_info(
+    ["imagenhub_Kolors_generation"],
+    "Kolors",
+    "https://huggingface.co/Kwai-Kolors/Kolors",
+    "Kolors is a large-scale text-to-image generation model based on latent diffusion",
+)
+
 # regist image edition models
 register_model_info(
     ["imagenhub_CycleDiffusion_edition"],

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', 'imagenhub_SD3_generation']  # 'playground_PlayGroundV2_generation'
+                           'playground_PlayGroundV2.5_generation', 'imagenhub_Kolors_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',