From ce592f8bcf2b261d96034993bc25a66e90006e99 Mon Sep 17 00:00:00 2001 From: myhloli Date: Fri, 6 Dec 2024 09:59:22 +0800 Subject: [PATCH] perf(model): optimize model initialization - Add condition to return existing model if already initialized - Improve efficiency by avoiding redundant model creation --- magic_pdf/model/sub_modules/model_init.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/magic_pdf/model/sub_modules/model_init.py b/magic_pdf/model/sub_modules/model_init.py index 0c4b0891..f4e94bee 100644 --- a/magic_pdf/model/sub_modules/model_init.py +++ b/magic_pdf/model/sub_modules/model_init.py @@ -102,10 +102,13 @@ class AtomModelSingleton: with self._lock: if key not in self._models: self._models[key] = atom_model_init(model_name=atom_model_name, **kwargs) + else: + return self._models[key] else: if key not in self._models: self._models[key] = atom_model_init(model_name=atom_model_name, **kwargs) - return self._models[key] + else: + return self._models[key] def atom_model_init(model_name: str, **kwargs):