Skip to content

Commit 8cf66b9

Browse files
authored
perf: Refine the Model Manager code (#3093)
1 parent 1db8577 commit 8cf66b9

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

apps/common/config/embedding_config.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,10 @@ def get_model(_id, get_model):
2323
model_instance = ModelManage.cache.get(_id)
2424
if model_instance is None:
2525
with _lock:
26-
model_instance = get_model(_id)
27-
ModelManage.cache.set(_id, model_instance, timeout=60 * 60 * 8)
26+
model_instance = ModelManage.cache.get(_id)
27+
if model_instance is None:
28+
model_instance = get_model(_id)
29+
ModelManage.cache.set(_id, model_instance, timeout=60 * 60 * 8)
2830
else:
2931
if model_instance.is_cache_model():
3032
ModelManage.cache.touch(_id, timeout=60 * 60 * 8)

0 commit comments

Comments
 (0)