yuanshengni commited on
Commit
8bd6620
1 Parent(s): a071819
Files changed (1) hide show
  1. model/model_manager.py +2 -2
model/model_manager.py CHANGED
@@ -84,13 +84,13 @@ class ModelManager:
84
  model_names = [model_A, model_B]
85
 
86
  with concurrent.futures.ThreadPoolExecutor() as executor:
87
- futures = [executor.submit(self.generate_image_vg, prompt, model) for model in model_names]
88
  results = [future.result() for future in futures]
89
  return results[0], results[1], model_names[0], model_names[1]
90
 
91
  def generate_video_vg_parallel(self, prompt, model_A, model_B):
92
  model_names = [model_A, model_B]
93
  with concurrent.futures.ThreadPoolExecutor() as executor:
94
- futures = [executor.submit(self.generate_image_vg, prompt, model) for model in model_names]
95
  results = [future.result() for future in futures]
96
  return results[0], results[1]
 
84
  model_names = [model_A, model_B]
85
 
86
  with concurrent.futures.ThreadPoolExecutor() as executor:
87
+ futures = [executor.submit(self.generate_video_vg, prompt, model) for model in model_names]
88
  results = [future.result() for future in futures]
89
  return results[0], results[1], model_names[0], model_names[1]
90
 
91
  def generate_video_vg_parallel(self, prompt, model_A, model_B):
92
  model_names = [model_A, model_B]
93
  with concurrent.futures.ThreadPoolExecutor() as executor:
94
+ futures = [executor.submit(self.generate_video_vg, prompt, model) for model in model_names]
95
  results = [future.result() for future in futures]
96
  return results[0], results[1]