vinesmsuic commited on
Commit
fa656b2
1 Parent(s): 2707b9d

updating model registry for DiT

Browse files
Files changed (1) hide show
  1. model/model_registry.py +7 -0
model/model_registry.py CHANGED
@@ -122,6 +122,13 @@ register_model_info(
122
  "StableCascade is built upon the Würstchen architecture and working at a much smaller latent space.",
123
  )
124
 
 
 
 
 
 
 
 
125
  # regist image edition models
126
  register_model_info(
127
  ["imagenhub_CycleDiffusion_edition"],
 
122
  "StableCascade is built upon the Würstchen architecture and working at a much smaller latent space.",
123
  )
124
 
125
+ register_model_info(
126
+ ["imagenhub_HunyuanDiT_generation"],
127
+ "HunyuanDiT",
128
+ "https://github.com/Tencent/HunyuanDiT",
129
+ "HunyuanDiT is a Powerful Multi-Resolution Diffusion Transformer with Fine-Grained Chinese Understanding",
130
+ )
131
+
132
  # regist image edition models
133
  register_model_info(
134
  ["imagenhub_CycleDiffusion_edition"],