adding CosXL and InfEdit

model/model_registry.py

@@ -158,6 +158,20 @@ register_model_info(
     "Plug-and-Play Diffusion Features for Text-Driven Image-to-Image Translation.",
 )
 
+register_model_info(
+    ["imagenhub_InfEdit_edition"],
+    "InfEdit",
+    "https://sled-group.github.io/InfEdit/",
+    "Inversion-Free Image Editing with Natural Language.",
+)
+
+register_model_info(
+    ["imagenhub_CosXLEdit_edition"],
+    "CosXLEdit",
+    "https://huggingface.co/stabilityai/cosxl",
+    "An instruction-based image editing model from SDXL.",
+)
+
 register_model_info(
     ["fal_stable-cascade_text2image"],
     "StableCascade",
@@ -205,6 +219,6 @@ models = ['imagenhub_LCM_generation','imagenhub_SDXLTurbo_generation','imagenhub
           'imagenhub_OpenJourney_generation','imagenhub_PixArtAlpha_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_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition', 'imagenhub_MagicBrush_edition', 'imagenhub_PNP_edition', 'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition',
           "fal_AnimateDiffTurbo_text2video", "fal_AnimateDiff_text2video",
           "videogenhub_LaVie_generation", "videogenhub_VideoCrafter2_generation", "videogenhub_ModelScope_generation"]