Spaces:
Sleeping
Sleeping
Gallinator commited on
Server: use forkserver as worker context
Browse files- 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 |
|