Skip to content

GitLab

Commit a9723c61 authored by myhloli's avatar myhloli
Browse files

fix(model): simplify model initialization logic

parent 878f3de0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
magic_pdf/model/sub_modules/model_init.py
View file @ a9723c61
...@@ -97,8 +97,7 @@ class AtomModelSingleton: ...@@ -97,8 +97,7 @@ class AtomModelSingleton:
key = (atom_model_name, layout_model_name, lang) key = (atom_model_name, layout_model_name, lang)
if key not in self._models: if key not in self._models:
self._models[key] = atom_model_init(model_name=atom_model_name, **kwargs) self._models[key] = atom_model_init(model_name=atom_model_name, **kwargs)
else: return self._models[key]
return self._models[key]
def atom_model_init(model_name: str, **kwargs): def atom_model_init(model_name: str, **kwargs):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment