Add Aurora

model/model_manager.py

@@ -63,7 +63,7 @@ class ModelManager:
     def generate_image_ig_parallel_anony(self, prompt, model_A, model_B, model_C, model_D):
         if model_A == "" and model_B == "" and model_C == "" and model_D == "":
             from .matchmaker import matchmaker
-            not_run = [20,21,22, 25,26, 30] #12,13,14,15,16,17,18,19,20,21,22, #23,24,
+            not_run = [20,21,22, 25,26, 30, 34] #12,13,14,15,16,17,18,19,20,21,22, #23,24,
             model_ids = matchmaker(num_players=len(self.model_ig_list), not_run=not_run)
             print(model_ids)
             model_names = [self.model_ig_list[i] for i in model_ids]

model/models/__init__.py

@@ -39,6 +39,7 @@ IMAGE_GENERATION_MODELS = [
                             "replicate_FLUX-1.1-pro_text2image",
                             'replicate_SD-v3.5-large_text2image',
                             'replicate_SD-v3.5-large-turbo_text2image',
+                            'other_Aurora_text2image',
                             ]
 
 VIDEO_GENERATION_MODELS = ['replicate_Zeroscope-v2-xl_text2video',