From 8cf66b9eca3f239e7a2b3762d32c2632bca89a19 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Thu, 15 May 2025 22:38:20 +0800 Subject: [PATCH] perf: Refine the Model Manager code (#3093) --- apps/common/config/embedding_config.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/apps/common/config/embedding_config.py b/apps/common/config/embedding_config.py index 03d585e2d..9f8a2f65a 100644 --- a/apps/common/config/embedding_config.py +++ b/apps/common/config/embedding_config.py @@ -23,8 +23,10 @@ class ModelManage: model_instance = ModelManage.cache.get(_id) if model_instance is None: with _lock: - model_instance = get_model(_id) - ModelManage.cache.set(_id, model_instance, timeout=60 * 60 * 8) + model_instance = ModelManage.cache.get(_id) + if model_instance is None: + model_instance = get_model(_id) + ModelManage.cache.set(_id, model_instance, timeout=60 * 60 * 8) else: if model_instance.is_cache_model(): ModelManage.cache.touch(_id, timeout=60 * 60 * 8)