Spaces:
Running
on
Zero
Running
on
Zero
Update KLing-v1.0
Browse files- model/matchmaker_video.py +1 -1
- model/model_manager.py +1 -1
model/matchmaker_video.py
CHANGED
@@ -106,7 +106,7 @@ def matchmaker_video(num_players, k_group=4):
|
|
106 |
ucb_scores[selected_player] = -float('inf')
|
107 |
|
108 |
excluded_players_1 = [7, 10]
|
109 |
-
excluded_players_2 = [6, 8, 9
|
110 |
excluded_players = excluded_players_1 + excluded_players_2
|
111 |
if selected_player in excluded_players_1:
|
112 |
for player in excluded_players:
|
|
|
106 |
ucb_scores[selected_player] = -float('inf')
|
107 |
|
108 |
excluded_players_1 = [7, 10]
|
109 |
+
excluded_players_2 = [6, 8, 9]
|
110 |
excluded_players = excluded_players_1 + excluded_players_2
|
111 |
if selected_player in excluded_players_1:
|
112 |
for player in excluded_players:
|
model/model_manager.py
CHANGED
@@ -113,7 +113,7 @@ class ModelManager:
|
|
113 |
|
114 |
root_dir = SSH_CACHE_OPENSOURCE
|
115 |
for name in model_names:
|
116 |
-
if "Runway-Gen3" in name or "Runway-Gen2" in name or "Pika-v1.0" in name
|
117 |
root_dir = SSH_CACHE_ADVANCE
|
118 |
elif "Pika-beta" in name:
|
119 |
root_dir = SSH_CACHE_PIKA
|
|
|
113 |
|
114 |
root_dir = SSH_CACHE_OPENSOURCE
|
115 |
for name in model_names:
|
116 |
+
if "Runway-Gen3" in name or "Runway-Gen2" in name or "Pika-v1.0" in name:
|
117 |
root_dir = SSH_CACHE_ADVANCE
|
118 |
elif "Pika-beta" in name:
|
119 |
root_dir = SSH_CACHE_PIKA
|