Add Sora-release

model/matchmaker_video.py

@@ -106,7 +106,7 @@ def matchmaker_video(num_players, k_group=4):
     ucb_scores[selected_player] = -float('inf') 
 
     excluded_players_1 = [7, 10]
-    excluded_players_2 = [6, 8, 9]
+    excluded_players_2 = [6, 8, 9, 13]
     excluded_players = excluded_players_1 + excluded_players_2
     if selected_player in excluded_players_1:
         for player in excluded_players:

model/model_manager.py

@@ -113,7 +113,7 @@ class ModelManager:
         
         root_dir = SSH_CACHE_OPENSOURCE
         for name in model_names:
-            if "Runway-Gen3" in name or "Runway-Gen2" in name or "Pika-v1.0" in name:
+            if "Runway-Gen3" in name or "Runway-Gen2" in name or "Pika-v1.0" in name or "Sora-release" in name:
                 root_dir = SSH_CACHE_ADVANCE
             elif "Pika-beta" in name:
                 root_dir = SSH_CACHE_PIKA      

model/models/__init__.py

@@ -55,6 +55,7 @@ VIDEO_GENERATION_MODELS = ['replicate_Zeroscope-v2-xl_text2video',
                             'other_Sora_text2video',
                             'replicate_Cogvideox-5b_text2video',
                             'other_KLing-v1.0_text2video',
+                            'other_Sora-release_text2video'
                            ]