boosting chances for new model

model/model_manager.py

@@ -5,7 +5,7 @@ import requests
 import io, base64, json, os
 import spaces
 from PIL import Image
-from .models import IMAGE_GENERATION_MODELS, IMAGE_EDITION_MODELS, VIDEO_GENERATION_MODELS, MUSEUM_UNSUPPORTED_MODELS, DESIRED_APPEAR_MODEL, load_pipeline
+from .models import IMAGE_GENERATION_MODELS, IMAGE_EDITION_MODELS, VIDEO_GENERATION_MODELS, MUSEUM_UNSUPPORTED_MODELS, DESIRED_APPEAR_MODEL, DESIRED_APPEAR_MODEL_CHANCE, load_pipeline
 from .fetch_museum_results import draw_from_imagen_museum, draw2_from_imagen_museum, draw_from_videogen_museum, draw2_from_videogen_museum
 from .pre_download import pre_download_all_models, pre_download_image_models_gen, pre_download_image_models_edit, pre_download_video_models_gen
 from transformers import AutoTokenizer, AutoModelForCausalLM
@@ -327,12 +327,22 @@ class ModelManager:
     def generate_video_vg_museum_parallel_anony(self, model_A, model_B):
         # Using list comprehension to get the difference between two lists
         picking_list = [item for item in self.model_vg_list if item not in self.excluding_model_list]
-        #picking_list = [item for item in picking_list if item not in self.desired_model_list]
+        
         if model_A == "" and model_B == "":
-            model_names = random.sample([model for model in picking_list], 2)
-
-            #override the random selection
-            #model_names[random.choice([0, 1])] = random.choice(self.desired_model_list)
+            # Filter desired_model_list to only include models that exist in picking_list
+            valid_desired_models = [m for m in self.desired_model_list if m in picking_list]
+            
+            # 50% (or DESIRED_APPEAR_MODEL_CHANCE) chance to include exactly one model from valid desired_model_list
+            if valid_desired_models and random.random() < DESIRED_APPEAR_MODEL_CHANCE:
+                # Pick one model from valid desired list
+                desired_model = random.choice(valid_desired_models)
+                # Pick one model from regular list, excluding desired models
+                regular_model = random.choice([m for m in picking_list if m not in valid_desired_models])
+                # Randomly determine order
+                model_names = [desired_model, regular_model] if random.random() < 0.5 else [regular_model, desired_model]
+            else:
+                # Pick two models from the regular picking list
+                model_names = random.sample([model for model in picking_list], 2)
         else:
             model_names = [model_A, model_B]
 

model/models/__init__.py

@@ -37,7 +37,8 @@ MAP_NAMES_IMAGENHUB = {}
 MAP_NAMES_VIDEOGENHUB = {"CogVideoX-2B": "CogVideoX", "CogVideoX-5B": "CogVideoX5B"}
 
 MUSEUM_UNSUPPORTED_MODELS = []
-DESIRED_APPEAR_MODEL = ['videogenhub_T2VTurbo_generation','fal_StableVideoDiffusion_text2video']
+DESIRED_APPEAR_MODEL = ['videogenhub_Mochi1_text2video']
+DESIRED_APPEAR_MODEL_CHANCE = 0.5
 
 ALL_MODELS = IMAGE_GENERATION_MODELS + IMAGE_EDITION_MODELS + VIDEO_GENERATION_MODELS