ffreemt commited on
Commit
4159401
1 Parent(s): df9248e

Fix concurrency_count to avoid OOM

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -473,7 +473,8 @@ with gr.Blocks(
473
  # CPU cpu_count=2 16G, model 7G
474
  # CPU UPGRADE cpu_count=8 32G, model 7G
475
 
476
- concurrency_count = max(psutil.virtual_memory().total / 10**9 // file_size - 1, 1)
 
477
  logger.info(f"{concurrency_count=}")
478
 
479
  block.queue(concurrency_count=concurrency_count, max_size=5).launch(debug=True)
 
473
  # CPU cpu_count=2 16G, model 7G
474
  # CPU UPGRADE cpu_count=8 32G, model 7G
475
 
476
+ _ = int(psutil.virtual_memory().total / 10**9 // file_size - 1)
477
+ concurrency_count = max(_, 1)
478
  logger.info(f"{concurrency_count=}")
479
 
480
  block.queue(concurrency_count=concurrency_count, max_size=5).launch(debug=True)