vinesmsuic commited on
Commit
8d30b3b
1 Parent(s): a40414f

Adding SD3

Browse files
model/model_registry.py CHANGED
@@ -94,6 +94,13 @@ register_model_info(
94
  "SDXL is a Latent Diffusion Model that uses two fixed, pretrained text encoders.",
95
  )
96
 
 
 
 
 
 
 
 
97
  register_model_info(
98
  ["imagenhub_PixArtAlpha_generation"],
99
  "PixArtAlpha",
 
94
  "SDXL is a Latent Diffusion Model that uses two fixed, pretrained text encoders.",
95
  )
96
 
97
+ register_model_info(
98
+ ["imagenhub_SD3_generation"],
99
+ "SD3",
100
+ "https://huggingface.co/blog/sd3",
101
+ "SD3 is a novel Multimodal Diffusion Transformer (MMDiT) model.",
102
+ )
103
+
104
  register_model_info(
105
  ["imagenhub_PixArtAlpha_generation"],
106
  "PixArtAlpha",
model/models/__init__.py CHANGED
@@ -9,7 +9,7 @@ from .videogenhub_models import load_videogenhub_model
9
  # 'playground_PlayGroundV2_generation', 'playground_PlayGroundV2.5_generation']
10
  IMAGE_GENERATION_MODELS = ['imagenhub_SDXLTurbo_generation','imagenhub_SDXL_generation', 'imagenhub_PixArtAlpha_generation', 'imagenhub_PixArtSigma_generation',
11
  'imagenhub_OpenJourney_generation','imagenhub_SDXLLightning_generation', 'imagenhub_StableCascade_generation', 'imagenhub_HunyuanDiT_generation',
12
- 'playground_PlayGroundV2.5_generation'] # 'playground_PlayGroundV2_generation'
13
  IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
14
  'imagenhub_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition',
15
  'imagenhub_MagicBrush_edition', 'imagenhub_PNP_edition',
 
9
  # 'playground_PlayGroundV2_generation', 'playground_PlayGroundV2.5_generation']
10
  IMAGE_GENERATION_MODELS = ['imagenhub_SDXLTurbo_generation','imagenhub_SDXL_generation', 'imagenhub_PixArtAlpha_generation', 'imagenhub_PixArtSigma_generation',
11
  'imagenhub_OpenJourney_generation','imagenhub_SDXLLightning_generation', 'imagenhub_StableCascade_generation', 'imagenhub_HunyuanDiT_generation',
12
+ 'playground_PlayGroundV2.5_generation', 'imagenhub_SD3_generation'] # 'playground_PlayGroundV2_generation'
13
  IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
14
  'imagenhub_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition',
15
  'imagenhub_MagicBrush_edition', 'imagenhub_PNP_edition',