ksort commited on
Commit
a40e735
·
1 Parent(s): 1e80fbd

Update KLing-v1.0

Browse files
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, 12]
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 or "KLing-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