add pixart sigma

model/model_registry.py

@@ -94,6 +94,13 @@ register_model_info(
     "Pixart-α consists of pure transformer blocks for latent diffusion.",
 )
 
+register_model_info(
+    ["imagenhub_PixArtSigma_generation"],
+    "PixArtSigma",
+    "https://github.com/PixArt-alpha/PixArt-sigma",
+    "Improved version of Pixart-α.",
+)
+
 register_model_info(
     ["imagenhub_SDXLLightning_generation"],
     "SDXL-Lightning",
@@ -230,7 +237,7 @@ register_model_info(
     
 
 models = ['imagenhub_LCM_generation','imagenhub_SDXLTurbo_generation','imagenhub_SDXL_generation',
-          'imagenhub_OpenJourney_generation','imagenhub_PixArtAlpha_generation','imagenhub_SDXLLightning_generation',
+          'imagenhub_OpenJourney_generation','imagenhub_PixArtAlpha_generation','imagenhub_PixArtSigma_generation','imagenhub_SDXLLightning_generation',
           'imagenhub_StableCascade_generation','imagenhub_PlaygroundV2_generation', 'fal_Playground-v25_generation', 'fal_stable-cascade_text2image',
           'imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
           'imagenhub_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition', 'imagenhub_MagicBrush_edition', 'imagenhub_PNP_edition', 'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition',

model/models/__init__.py

@@ -3,7 +3,7 @@ from .playground_api import load_playground_model
 from .fal_api_models import load_fal_model
 from .videogenhub_models import load_videogenhub_model
 
-IMAGE_GENERATION_MODELS = ['imagenhub_LCM_generation','imagenhub_SDXLTurbo_generation','imagenhub_SDXL_generation', 'imagenhub_PixArtAlpha_generation',
+IMAGE_GENERATION_MODELS = ['imagenhub_LCM_generation','imagenhub_SDXLTurbo_generation','imagenhub_SDXL_generation', 'imagenhub_PixArtAlpha_generation', 'imagenhub_PixArtSigma_generation',
                             'imagenhub_OpenJourney_generation','imagenhub_SDXLLightning_generation', 'imagenhub_StableCascade_generation',
                             'playground_PlayGroundV2_generation', 'playground_PlayGroundV2.5_generation']
 IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',