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)))