khang119966 commited on
Commit
b0a9ec2
·
verified ·
1 Parent(s): a623f18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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