Gallinator commited on
Commit
2e7ff4d
·
unverified ·
1 Parent(s): b4b8524

Server: use forkserver as worker context

Browse files
Files changed (1) hide show
  1. app/server.py +2 -1
app/server.py CHANGED
@@ -5,6 +5,7 @@ import traceback
5
  from concurrent.futures import ProcessPoolExecutor
6
  from contextlib import asynccontextmanager
7
  from functools import partial
 
8
  import numpy as np
9
  import torch
10
  from fastapi import FastAPI, HTTPException
@@ -150,7 +151,7 @@ async def lifespan(app: FastAPI):
150
  settings = AppSettings()
151
  app = FastAPI(lifespan=lifespan)
152
  openfoam_cmd = f'{settings.openfoam_dir}/etc/openfoam'
153
- app.process_pool = ProcessPoolExecutor()
154
  app.models = {}
155
 
156
 
 
5
  from concurrent.futures import ProcessPoolExecutor
6
  from contextlib import asynccontextmanager
7
  from functools import partial
8
+ from multiprocessing import get_context
9
  import numpy as np
10
  import torch
11
  from fastapi import FastAPI, HTTPException
 
151
  settings = AppSettings()
152
  app = FastAPI(lifespan=lifespan)
153
  openfoam_cmd = f'{settings.openfoam_dir}/etc/openfoam'
154
+ app.process_pool = ProcessPoolExecutor(mp_context=get_context("forkserver"))
155
  app.models = {}
156
 
157