Adding ultraedit

model/model_registry.py

@@ -194,6 +194,13 @@ register_model_info(
     "An instruction-based image editing model from SDXL.",
 )
 
+register_model_info(
+    ["imagenhub_UltraEdit_edition"],
+    "UltraEdit",
+    "https://ultra-editing.github.io/",
+    "Instruction-based Fine-Grained Image Editing at Scale.",
+)
+
 register_model_info(
     ["fal_stable-cascade_text2image"],
     "StableCascade",
@@ -250,3 +257,10 @@ register_model_info(
     "A community-driven opensource implementation of Sora.",
 )
     
+register_model_info(
+    ["videogenhub_T2VTurbo_generation"],
+    "T2V-Turbo",
+    "https://github.com/Ji4chenLi/t2v-turbo",
+    "Video Consistency Model with Mixed Reward Feedback.",
+)
+    

model/models/__init__.py

@@ -13,7 +13,7 @@ IMAGE_GENERATION_MODELS = ['imagenhub_SDXLTurbo_generation','imagenhub_SDXL_gene
 IMAGE_EDITION_MODELS = ['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']
+                        'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition', 'imagenhub_UltraEdit_edition']
 VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
                            'fal_AnimateDiffTurbo_text2video',
                            'videogenhub_LaVie_generation',