Эх сурвалжийг харах

fix:'title_aided_config' is not defined

myhloli 4 сар өмнө
parent
commit
92d43c7b23

+ 2 - 1
mineru/backend/vlm/token_to_middle_json.py

@@ -12,7 +12,8 @@ from mineru.version import __version__
 
 
 heading_level_import_success = False
 heading_level_import_success = False
 llm_aided_config = get_llm_aided_config()
 llm_aided_config = get_llm_aided_config()
-if llm_aided_config and llm_aided_config.get('title_aided', {}).get('enable', False):
+title_aided_config = llm_aided_config.get('title_aided', {})
+if llm_aided_config and title_aided_config.get('enable', False):
     try:
     try:
         from mineru.utils.llm_aided import llm_aided_title
         from mineru.utils.llm_aided import llm_aided_title
         from mineru.backend.pipeline.model_init import AtomModelSingleton
         from mineru.backend.pipeline.model_init import AtomModelSingleton