ksort commited on
Commit
c0eefa5
·
1 Parent(s): b2fe563

Add GPT-4o model

Browse files
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 = [20,21,22, 25,26, 30, 34] #12,13,14,15,16,17,18,19,20,21,22, #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, 25,26, 30, 34, 35] #12,13,14,15,16,17,18,19,20,21,22, #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]
model/models/__init__.py CHANGED
@@ -40,6 +40,7 @@ IMAGE_GENERATION_MODELS = [
40
  'replicate_SD-v3.5-large_text2image',
41
  'replicate_SD-v3.5-large-turbo_text2image',
42
  'other_Aurora_text2image',
 
43
  ]
44
 
45
  VIDEO_GENERATION_MODELS = ['replicate_Zeroscope-v2-xl_text2video',
 
40
  'replicate_SD-v3.5-large_text2image',
41
  'replicate_SD-v3.5-large-turbo_text2image',
42
  'other_Aurora_text2image',
43
+ 'other_GPT-4o_text2image',
44
  ]
45
 
46
  VIDEO_GENERATION_MODELS = ['replicate_Zeroscope-v2-xl_text2video',