Add AURORA

model/model_registry.py

@@ -321,6 +321,16 @@ register_model_info(
     "image_edition"
 )
 
+register_model_info(
+    ["imagenhub_AURORA_edition"],
+    "AURORA",
+    "https://aurora-editing.github.io/",
+    "AURORA (Action Reasoning Object Attribute) enables training an instruction-guided image editing model that can perform action and reasoning-centric edits.",
+    "MIT",
+    "McGill NLP",
+    "image_edition"
+)
+
 register_model_info(
     ["fal_stable-cascade_text2image"],
     "StableCascade",

model/models/__init__.py

@@ -15,7 +15,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_UltraEdit_edition']
+                        'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition', 'imagenhub_UltraEdit_edition', 'imagenhub_AURORA_edition']
 VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
                            'fal_AnimateDiffTurbo_text2video',
                            #'videogenhub_LaVie_generation',