From 76e759489518b22d7e3e78b7a27740072567396f Mon Sep 17 00:00:00 2001 From: hailin Date: Tue, 5 Aug 2025 17:08:08 +0800 Subject: [PATCH] . --- app/main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/app/main.py b/app/main.py index 761c096..27aeee5 100644 --- a/app/main.py +++ b/app/main.py @@ -182,7 +182,7 @@ logger.info("Using SAFE_MIN_FREE_MB = %d MB", SAFE_MIN_FREE_MB) def _warm_worker(t, q): try: - _ = model.encode(t, return_dense=True, num_processes=1) + _ = model.encode(t, return_dense=True) q.put("ok") except Exception as e: q.put(str(e)) @@ -216,7 +216,7 @@ class EmbeddingRequest(BaseModel): def _worker(t, q): try: # out = model.encode(t, return_dense=True) # GPU or CPU 均安全 - out = model.encode(t, return_dense=True, num_processes=1) + out = model.encode(t, return_dense=True) q.put(("ok", out)) except Exception as e: q.put(("err", str(e)))