ksort commited on
Commit
766eee9
1 Parent(s): 796d9ad

Fix gpu bug

Browse files
Files changed (1) hide show
  1. model/model_manager.py +1 -1
model/model_manager.py CHANGED
@@ -63,7 +63,7 @@ class ModelManager:
63
  def generate_image_ig_parallel_anony(self, prompt, model_A, model_B, model_C, model_D):
64
  if model_A == "" and model_B == "" and model_C == "" and model_D == "":
65
  from .matchmaker import matchmaker
66
- not_run = [] #12,13,14,15,16,17,18,19,20,21,22, 25,26 #23,24,
67
  model_ids = matchmaker(num_players=len(self.model_ig_list), not_run=not_run)
68
  print(model_ids)
69
  model_names = [self.model_ig_list[i] for i in model_ids]
 
63
  def generate_image_ig_parallel_anony(self, prompt, model_A, model_B, model_C, model_D):
64
  if model_A == "" and model_B == "" and model_C == "" and model_D == "":
65
  from .matchmaker import matchmaker
66
+ not_run = [20,21,22] #12,13,14,15,16,17,18,19,20,21,22, 25,26 #23,24,
67
  model_ids = matchmaker(num_players=len(self.model_ig_list), not_run=not_run)
68
  print(model_ids)
69
  model_names = [self.model_ig_list[i] for i in model_ids]