Spaces:
Running
on
Zero
Running
on
Zero
yuanshengni
commited on
Commit
•
8bd6620
1
Parent(s):
a071819
fix vg
Browse files- 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.
|
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.
|
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]
|