Spaces:
Running
on
Zero
Running
on
Zero
Fix gpu bug
Browse files- 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]
|