Spaces:
Running
on
Zero
Running
on
Zero
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]
|