vinesmsuic commited on
Commit
98bef81
1 Parent(s): 7757ac2

revert to normal selection

Browse files
Files changed (1) hide show
  1. model/model_manager.py +2 -2
model/model_manager.py CHANGED
@@ -195,12 +195,12 @@ class ModelManager:
195
  def generate_video_vg_museum_parallel_anony(self, model_A, model_B):
196
  # Using list comprehension to get the difference between two lists
197
  picking_list = [item for item in self.model_vg_list if item not in self.excluding_model_list]
198
- picking_list = [item for item in picking_list if item not in self.desired_model_list]
199
  if model_A == "" and model_B == "":
200
  model_names = random.sample([model for model in picking_list], 2)
201
 
202
  #override the random selection
203
- model_names[random.choice([0, 1])] = random.choice(self.desired_model_list)
204
  else:
205
  model_names = [model_A, model_B]
206
 
 
195
  def generate_video_vg_museum_parallel_anony(self, model_A, model_B):
196
  # Using list comprehension to get the difference between two lists
197
  picking_list = [item for item in self.model_vg_list if item not in self.excluding_model_list]
198
+ #picking_list = [item for item in picking_list if item not in self.desired_model_list]
199
  if model_A == "" and model_B == "":
200
  model_names = random.sample([model for model in picking_list], 2)
201
 
202
  #override the random selection
203
+ #model_names[random.choice([0, 1])] = random.choice(self.desired_model_list)
204
  else:
205
  model_names = [model_A, model_B]
206