gmakstutis/openrag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

spawn fix

Browse source
This commit is contained in:
phact 2025-09-06 00:45:10 -04:00
parent be77cac353
commit 174823f5d4
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/main.py
View file

@ -14,6 +14,10 @@ import subprocess
from functools import partial from functools import partial
from starlette.applications import Starlette from starlette.applications import Starlette
from starlette.routing import Route from starlette.routing import Route
# Set multiprocessing start method to 'spawn' for CUDA compatibility
multiprocessing.set_start_method("spawn", force=True)
from utils.process_pool import process_pool from utils.process_pool import process_pool
import torch import torch
@ -50,8 +54,6 @@ from api import (
settings, settings,
) )
# Set multiprocessing start method to 'spawn' for CUDA compatibility
multiprocessing.set_start_method("spawn", force=True)
logger.info( logger.info(
"CUDA device information", "CUDA device information",