From c6c629f376632060a4c312b51111b6a3d9392906 Mon Sep 17 00:00:00 2001 From: hailin Date: Tue, 5 Aug 2025 17:12:49 +0800 Subject: [PATCH] . --- app/main.py | 16 +++++----------- 1 file changed, 5 insertions(+), 11 deletions(-) diff --git a/app/main.py b/app/main.py index 27aeee5..41bc936 100644 --- a/app/main.py +++ b/app/main.py @@ -192,18 +192,12 @@ def _warm_worker(t, q): def warm_up(): logger.info("Warm-up on %s", DEVICE) try: - texts = ["warmup"] - q = mp.Queue() - p = mp.Process(target=_warm_worker, args=(texts, q)) - p.start() - p.join(timeout=60) - - if not q.empty() and q.get() == "ok": - logger.info("Warm-up complete.") - else: - logger.warning("Warm-up failed or timed out.") + _ = model.encode([ + "This is a warmup sentence used to initialize CUDA kernels and avoid latency spikes." + ], return_dense=True) + logger.info("Warm-up complete.") except Exception as e: - logger.warning("Warm-up exception: %s", e) + logger.warning("Warm-up failed: %s", e)