Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -532,7 +532,10 @@ def generate_video(image, prompt, max_tokens):
|
|
| 532 |
|
| 533 |
# Dùng multiprocessing
|
| 534 |
# with multiprocessing.Pool(processes=20) as pool:
|
| 535 |
-
with ProcessPoolExecutor(max_workers=20) as pool:
|
|
|
|
|
|
|
|
|
|
| 536 |
text_imgs = pool.map(generate_text_img_wrapper, params_for_text)
|
| 537 |
hidden_imgs = pool.map(generate_hidden_img_wrapper, params_for_hidden)
|
| 538 |
|
|
|
|
| 532 |
|
| 533 |
# Dùng multiprocessing
|
| 534 |
# with multiprocessing.Pool(processes=20) as pool:
|
| 535 |
+
# with ProcessPoolExecutor(max_workers=20) as pool:
|
| 536 |
+
ctx = multiprocessing.get_context()
|
| 537 |
+
ctx.Process(target=lambda: None).daemon = False
|
| 538 |
+
with ctx.Pool(processes=20) as pool:
|
| 539 |
text_imgs = pool.map(generate_text_img_wrapper, params_for_text)
|
| 540 |
hidden_imgs = pool.map(generate_hidden_img_wrapper, params_for_hidden)
|
| 541 |
|